[Do it! 알고리즘] K번째 수 구하기(백준 11004)

Do it! 알고리즘 코딩테스트

4-4. 퀵 정렬

정의

퀵 정렬(quick start)은 기준값(pivot)을 선정해 해당 값보다 작은 데이터와 큰 데이터로 분류하는 것을 반복해 정렬하는 알고리즘

퀵 정렬 과정

1. 데이터를 분할하는 pivot 선정
2. pivot을 기준으로 다음 a~e 과정을 거쳐 데이터를 2개의 집합으로 분리
   2.a start가 가리키는 데이터가 pivot이 가리키는 데이터보다 작으면 start를 오른쪽으로 1칸 이동
   2.b end가 가리키는 데이터가 pivot이 가리키는 데이터보다 크면 end를 왼쪽으로 1칸 이동
   2.c start가 가리키는 데이터가 pivot이 가리키는 데이터보다 크고, end가 가리키는 데이터가 pivot이 가리키는 데이터보다 작으면 start, end 가 가리키는 데이터를 swap하고 start는 오른쪽, end는 왼쪽으로 1칸씩 이동
   2.d start와 end가 만날 때까지 2.a ~ 2.c를 반복
   2.e start와 end가 만나면 만난 지점에서 가리키는 데이터와 pivot이 가리키는 데이터를 비교하여 pivot이 가리키는 데이터가 크면 만난 지점의 오 른쪽에, 작으면 만난 지점의 왼쪽에 pivot이 가리키는 데이터를 삽입
3. 분리 집합에서 각각 다시 pivot을 선정
4. 분리 집합이 1개 이하가 될 때까지 1~3을 반복

백준 11004

코드

• Sort 함수 풀이

N, K = map(int,input().split())

nums = list(map(int, input().split()))
nums.sort()

print(nums[K-1])

• 퀵 정렬 풀이

N, K = (map(int,input().split()))
nums = list(map(int,input().split()))

def solution(N, K, nums):
    answer = 0
    quicksort(0, N-1, K-1)
    answer = nums[K-1]
    return answer

def quicksort(start, end, K):
    global nums
    if (start < end):

        pivot = partition(start, end)

        if (pivot < K):
            quicksort(pivot+1, end, K)
        elif (K < pivot):
            quicksort(start, pivot-1, K)
        else:
            return


def partition(start, end):
    global nums

    #데이터가 2개인 경우 바로 비교해서 정렬
    if(start+1 == end):
        if nums[start] > nums[end]:
            swap(start, end)
        return end

    pivot = (start + end)//2
    
    swap(start, pivot)
    pivot = start
    i = start + 1
    j = end
    while(i<=j):
        #피벗보다 큰 수가 나올 때 까지 i 증가
        while nums[pivot]> nums[i] and i<len(nums)-1:
            i+=1
        #피벗보다 작은 수가 나올때 까지 j 감소
        while(nums[pivot] < nums[j] and j>0):
            j-=1
        if i <= j:
            swap(i, j)
            i += 1
            j -= 1
            
    temp = nums[pivot]
    nums[start] = nums[j]
    nums[j] = temp
    
    return j



def swap(i,j):
    global nums
    nums[i], nums[j] = nums[j], nums[i]


print(solution(N, K, nums))

Reference

• Do it! 알고리즘 코딩테스트 책
• 백준 11004 파이썬
• https://dingx2-story.tistory.com/71