퀵 정렬 (Quick Sort) 알고리즘 (with Python3)

퀵 정렬

퀵 정렬(Quick Sort)은 병합 정렬(Merge Sort)과 비슷하게 재귀 호출을 하여 정렬을 한다.

다만 퀵 정렬은 기준점과 미리 비교를 한다는 차이점이 있다.

Credit. https://medium.com/karuna-sehgal/a-quick-explanation-of-quick-sort-7d8e2563629b

코딩

아래 코드는 unittest 기반으로 작성되었다.

일반적인 퀵 정렬

# -*- coding: utf-8 -*-

import unittest


class Exam11(unittest.TestCase):

    @classmethod
    def setUp(cls):
        pass

    def test_quick_sort(self):
        """일반적인 퀵 정렬"""
        def _quick_sort(data, start, end):
            # 종료 조건: 정렬을 할 원소가 1개 이하인 경우
            if start >= end:
                return

            # 피봇 설정 (편의상 맨 마지막 원소를 피봇으로 지정)
            pivot = data[end]
            left = start

            # 왼쪽은 피봇보다 작은 수가 위치한다.
            # 오른쪽은 피봇보다 큰 수가 위치한다.
            for right in range(start, end):
                if data[right] <= pivot:
                    # data[right], data[left] = data[left], data[right] 와 동일
                    tmp = data[right]
                    data[right] = data[left]
                    data[left] = tmp

                    left += 1

            # 맨 마지막에 위치한 피봇과 위치를 바꿔준다.
            # [피봇보다 작은 수들] + [피봇] + [피봇보다 큰 수]
            data[end], data[left] = data[left], data[end]

            _quick_sort(data, start, left - 1)
            _quick_sort(data, left + 1, end)

        data = [6, 8, 3, 9, 10, 1, 2, 4, 7, 5]
        _quick_sort(data, 0, len(data) - 1)
        self.assertEqual([1, 2, 3, 4, 5, 6, 7, 8, 9, 10], data)


if __name__ == "__main__":
    unittest.main()

파이썬을 활용한 퀵 정렬

# -*- coding: utf-8 -*-

import unittest


class Exam11(unittest.TestCase):

    @classmethod
    def setUp(cls):
        pass

    def test_quick_sort_with_python(self):
        def _quick_sort(data):
            # 종료 조건
            n = len(data)
            if n <= 1:
                return data

            # 피봇 값은 어떤값으로 하든 상관없음. 편의상 맨 마지막을 피봇으로 지정
            pivot = data[-1]
            group1 = list()  # 피봇보다 작은 값
            group2 = list()  # 피봇보다 큰 값

            for num in data[:-1]:
                if num < pivot:
                    group1.append(num)
                else:
                    group2.append(num)

            return _quick_sort(group1) + [pivot] + _quick_sort(group2)

        data = [4, 2, 6, 5, 3, 9]
        self.assertEqual([2, 3, 4, 5, 6, 9], _quick_sort(data))


if __name__ == "__main__":
    unittest.main()

시간 복잡도는 병합 정렬과 똑같이 분할 정복을 통해 수행하기 때문에 O(N * logN) 이다.

다만, 대부분은 O(NlogN)이지만... 최악의 경우 O(N^2) 까지 갈 수 있다는 것을 알아두자.

 

다른 방법의 정렬 알고리즘은 아래를 참고한다.

• 2020/03/21 - [Algorithm] - 선택 정렬 (Selection Sort) 알고리즘 (with Python3)
• 2020/03/21 - [Algorithm] - 삽입 정렬 (Insertion Sort) 알고리즘 (with Python3)
• 2020/03/25 - [Algorithm] - 병합 정렬 (Merge Sort) 알고리즘 (with Python3)
• 2020/03/25 - [Algorithm] - 버블 정렬 (Bubble Sort) 알고리즘 (with Python3)

Reference

• https://thebook.io/006935/part03/ch11/03-01/