LeetCode #75 颜色分类 三向切分快速排序

LeetCode #75 颜色分类

题目描述

给定一个包含红色、白色和蓝色，一共 n 个元素的数组，原地对它们进行排序，使得相同颜色的元素相邻，并按照红色、白色、蓝色顺序排列。

此题中，我们使用整数 0、 1 和 2 分别表示红色、白色和蓝色。

注意:
不能使用代码库中的排序函数来解决这道题。

示例:

输入: [2,0,2,1,1,0]
输出: [0,0,1,1,2,2]

进阶：

• 一个直观的解决方案是使用计数排序的两趟扫描算法。
  首先，迭代计算出0、1 和 2 元素的个数，然后按照0、1、2的排序，重写当前数组。
• 你能想出一个仅使用常数空间的一趟扫描算法吗？

思路分析

这就是发明最短路径算法迪杰斯特拉Dijkstra大佬提出的荷兰国旗问题，敬礼！这个算法其实就是对快排的一种优化：在数组中存在大量重复数据的时候，快排会退化。这时候用三指针来将数组划分成 “比参照物大的元素”、“比参照物小的元素”和“等于参照物的元素” 三部分，再递归的时候只递归大的和小的元素，相等的不加入，这样就可以加速快排
【算法理解】—— 快速排序（三向切分）

class Solution:
    def sortColors(self, nums: List[int]) -> None:
        """
        Do not return anything, modify nums in-place instead.
        """
        # 用三个指针维护数组
        cur = 0; p1 = 0; p2 = len(nums) - 1
        while (cur <= p2):
            if nums[cur] == 0:
                # 比参考物小，往左移，cur 和左指针往后移动一位
                nums[cur], nums[p1] = nums[p1], nums[cur]
                cur += 1; p1 += 1
            elif nums[cur] == 1:
                # 跟参考物一样大，不动它，cur 往后移动一位
                cur += 1
            else:
                # 比参考物大，往右移，这时不知道从右边传过来的数是大是小，所以 cur 不动，右指针往左移动一位
                nums[cur], nums[p2] = nums[p2], nums[cur]
                p2 -= 1
        
        return nums

时间复杂度：$(N)$：只进行了一次遍历
空间复杂度：$(1)$：只用到了3个指针