카테고리 없음

퀵 정렬 - Python 일반적인 경우 & 간략화된 코드

아코20 2022. 7. 23. 19:30

일반적인 방식

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 <= 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] # 파이썬에서 swap
    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)

 

파이썬의 장점을 살린 방식

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:] # array의 1번째 항 부터 뒷부분만
    
    left_side = [x for x in tail if x <= pivot]
	right_side = [x for x in tain if x > pivot]
    
    return quick_sort(left_side) + [pivot] + quick_sort(right_side)
    
print(quick_sort(array))