본문 바로가기
알고리즘 풀이/정렬 알고리즘

정렬 알고리즘_ 퀵 정렬

by developer jini 2022. 12. 29.
728x90
array = [5, 7, 9, 0, 3, 1, 6, 2, 4, 8]


def quick_sort(array, start, end):
    if start >= end:  # 원소가 1개인 경우 종료 - 종료조건
        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)
728x90

댓글