[이코테]정렬 코드 정리

선택 정렬

array = [7, 5, 9, 0, 3, 1, 6, 2, 4, 8]

    for i in range(len(array)):
        min_index = i
        for j in range(i+1, len(array)):
            if array[min_index] > array[j]:
                min_index = j
        array[i], array[min_index] = array[min_index], array[i]

    print(array)

배열 인덱스 첫번째부터 i로 지정

그 이후 배열에서 최소값을 찾아 해당 인덱스를 j로 지정한 뒤

스와프(swap)를 통해 자리를 바꿔준다.

즉, 배열의 최소값을 맨 앞으로 이동

이를 반복하여 맨 앞에서부터 차례대로 하나씩 지정

 

 

삽입 정렬

array = [7, 5, 9, 0, 3, 1, 6, 2, 4, 8]

    for i in range(1, len(array)):
        for j in range(i, 0, -1):
            if array[j] < array[j - 1]:
                array[j], array[j-1] = array[j-1], array[j]
            else:
                break
    print(array)

j를 통해 인덱스 i부터 1까지 감소하며 반복한다.

if문을 통해 앞에 있는 요소보다 작으면 한 칸씩 앞으로 이동한다.

 

 

퀵 정렬

array = [5, 7, 9, 0, 3, 1, 6, 2, 4, 8]
    
def quick_sort(array, start, end):
    if start >= end:
        return
    pivot = start
    left = start + 1
    right = end
    while left <= right:
        while left <= end and array[left] <= array[pivot]:
            left += 1
        while right > start and array[right] >= array[pivot]:
            right -= 1
        if left > right:
            array[right], array[pivot] = array[pivot], array[right]
        else:
            array[left], array[right] = array[right], array[left]
    quick_sort(array, start, right-1)
    quick_sort(array, right+1, end)

quick_sort(array, 0, len(array)-1)
print(array)

if start >= end

원소가 1개인 경우 종료

 

pivot을 기준으로 left와 right를 지정

pivot은 첫 번째 원소로 지정

 

피벗보다 큰 데이터를 찾을 때까지 반복

left가 end인 리스트 끝을 넘어가지 않도록, array[left] 값이 array[pivot]의 값보다 작다면 left의 수를 계속 늘려나간다

계속해서 오른쪽으로 이동한다. 

 

피벗보다 작은 데이터를 찾을 때까지 반복

위와 동일

 

if문을 통해

엇갈렸다면 작은 데이터와 피벗을 교체

array[left]는 큰 값을 찾고, array[right]는 작은 값을 찾기 때문에 array[right]가 더 작은 데이터를 가지게 됨

그래서 array[right]와 pivot을 교체함

 

엇갈리지 않았다면 작은 데이터와 큰 데이터를 교체

기존 array[left]와 array[right]가 서로 교체됨

 

분할 이후 왼쪽 부분과 오른쪽 부분에서 각각 정렬 수행 - 재귀함수로 구현

 

 

파이썬의 장점을 살린 퀵 정렬 소스코드

array = [5, 7, 9, 0, 3, 1, 6, 2, 4, 8]

def quick_sort(array):
    if len(array) <= 1:
        return array
        
    pivot = array[0]
    tail = array[1:]
    
    left_side = [x for x in tail if x <= pivot]
    right_side = [x for x in tail if x > pivot]
    
    return quick_sort(left_side) + [pivot] + quick_sort(right_side)

print(quick_sort(array))

리스트가 하나 이하의 원소만을 담고 있다면 종료

 

피벗은 첫 번째 원소로 지정

tail에 피벗을 제외한 리스트를 지정

 

각각 분할된 왼쪽, 오른쪽 부분

분할 이후 왼쪽 부분과 오른쪽 부분에서 각각 정렬을 수행하고, 전체 리스트를 반환

 

 

계수 정렬

array = [7, 5, 9, 0, 3, 1, 6, 2, 9, 1, 4, 8, 0, 5, 2]

count = [0] * (max(array)+1)

for i in range(len(array)):
    count[array[i]] += 1

for i in range(len(count)):
    for j in range(count[i]):
        print(i, end=' ')

count 리스트에 각 데이터에 해당하는 인덱스의 값 증가

 

count 리스트의 인덱스 값(array 리스트에 들어있던 원소)을 i

count 리스트의 원소 값(array 리스트에 등장한 횟수)를 j

i를 j만큼 반복하여 출력

 

 

<출처>

이것이 취업을 위한 코딩 테스트다 with 파이썬