颜色分类

题目

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

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

示例 1：

输入：nums = [2,0,2,1,1,0]
输出：[0,0,1,1,2,2]

示例 2：

输入：nums = [2,0,1]
输出：[0,1,2]

示例 3：

输入：nums = [0]
输出：[0]

示例 4：

输入：nums = [1]
输出：[1]

提示：

n == nums.length
1 <= n <= 300
nums[i] 为 0、1 或 2

进阶：

你可以不使用代码库中的排序函数来解决这道题吗？
你能想出一个仅使用常数空间的一趟扫描算法吗？

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/sort-colors
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

解题思路

排序函数

0、1、2 刚好是按照题目要求的红色、白色、蓝色顺序，升序排列数组即可得到。

优化

利用快速排序的特点，逐一对数组元素进行判断归类，由于题目要求原地排序，于是采取两两交换的方法。

为实现一趟扫描数组得到结果，需要特别地设置数组两边的边界， 0 一定是在最左边部分，从左边第一个位置开始，同理，出现 2 也是一定从最右边部分的第一个位置开始， 1 出现在数组的中间部分。所以两边初始设置指针 left、 right来控制边界移动位置。

1. 元素为 0 时：保证 0 从最左边出现，交换此时 i 与 left 所指位置上的元素，left 指向 0 再向右移动，方便下次交换；
2. 元素为 1 时：不影响 0 和 2 出现的位置，i 继续向前遍历；
3. 元素为 2 时：保证 2 从最右边出现，交换此时 i 与 right 所指位置上的元素，right 指向 2 再向左移动，当遇到与 i 相同时，表示结束。

提交代码

排序函数

class Solution:
    def sortColors(self, nums: List[int]) -> None:
        """
        Do not return anything, modify nums in-place instead.
        """
        nums.sort()
        

结果

优化

class Solution:
    def sortColors(self, nums: List[int]) -> None:
        """
        Do not return anything, modify nums in-place instead.
        """
        left = 0
        right = len(nums)
        i = 0
        while i < right:
            if nums[i] == 0:
                nums[i],nums[left] = nums[left],nums[i]
                left += 1
                i += 1
            elif nums[i] == 1:
                i += 1
            else:
                right -= 1
                nums[i],nums[right] = nums[right],nums[i] 
                 

结果

总结

1. 优化时首先直接采用 for 循环，导致只能保证最左边的元素为 0 ，出现的结果 1 和 2 往往是颠倒的顺序，没有将 i 和 right 的关系联系起来；
2. while循环的优势在于方便控制 i 的变化，分三种情况可以看出，当出现 2 时 i 不会递增；
3. 双指针甚至多指针的思维实践更加熟练，较之前有进步。