【数组】力扣75题：颜色分类

建议在看题目之前先了解数组的具体知识点，可以看这里：
算法基础（二）：数组知识点及题型讲解。

其它题目：
【数组】力扣283题：移动0
【数组】力扣27题：移除元素
【数组】力扣26题：删除有序数组中的重复项
【数组】力扣80题：删除有序数组中的重复项 II
【数组】力扣485题：最大连续 1 的个数

力扣75题：颜色分类

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

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

必须在不使用库内置的 sort 函数的情况下解决这个问题。

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

来源：力扣（LeetCode）75

解法一：最简单，直接计算

直接计算每种颜色的个数然后重排数组

class Solution:
    def sortColors(self, nums: List[int]) -> None:
        """
        Do not return anything, modify nums in-place instead.
        """
        a = nums.count(0)
        b = nums.count(1)
        c = nums.count(2)
        nums[0:a] = a*[0]
        nums[a:a+b] = b*[1]
        nums[a+b:] = c*[2]

执行用时：36 ms, 在所有 Python3 提交中击败了72.15%的用户
内存消耗：14.9 MB, 在所有 Python3 提交中击败了46.57%的用户

也可以把c换掉，用len(nums)-a-b代替。

解法二：单指针法

接下来讲的这几种都是官方解法。

对数组进行两次遍历。第一次遍历，将数组中所有的 0 交换到数组的头部。第二次遍历，将数组中所有的 1 交换到头部的 0 之后。
时间复杂度O(N)，空间复杂度O(1)

class Solution:
    def swap(nums, a, b):
        c = nums[a]
        nums[a] = nums[b]
        nums[b] = c
        return nums

    def sortColors(self, nums: List[int]) -> None:
        # 单指针法
        head = 0
        for i in range(len(nums)):
            if nums[i] == 0:
                Solution.swap(nums, head, i)
                head += 1
        for i in range(head, len(nums)):
            if nums[i] == 1:
                Solution.swap(nums, head, i)
                head += 1

执行用时：32 ms, 在所有 Python3 提交中击败了90.01%的用户
内存消耗：15 MB, 在所有 Python3 提交中击败了14.69%的用户

解法三：0，1双指针法

仅使用一次遍历：额外使用一个指针，即使用两个指针分别用来交换 0 和 1。

class Solution:
    def swap(nums, a, b):
        c = nums[a]
        nums[a] = nums[b]
        nums[b] = c
        return nums

    def sortColors(self, nums: List[int]) -> None:
        # 双指针法
        p0 = p1 = 0
        for i in range(len(nums)):
            if nums[i] == 0:
                Solution.swap(nums, i, p0)
                if p0 < p1:  # 0后面是1，替换了0过来，就会代替掉1，1就到了i的位置，所以要再交换i和p1
                    Solution.swap(nums, i, p1)
                p0 += 1
                p1 += 1   
            elif nums[i] == 1:
                Solution.swap(nums, i, p1)
                p1 += 1         

执行用时：32 ms, 在所有 Python3 提交中击败了90.01%的用户
内存消耗：14.9 MB, 在所有 Python3 提交中击败了79.12%的用户

解法四：0，2双指针法

这里要注意，交换了2以后，nums[i]可能是0，而这时候还要再判断是否为0，如果是，要交换0。

class Solution:
    def swap(nums, a, b):
        c = nums[a]
        nums[a] = nums[b]
        nums[b] = c
        return nums

    def sortColors(self, nums: List[int]) -> None:
        # 双指针法
        n = len(nums)
        p0, p2, i = 0, n-1, 0
        while i <= p2:
            while i <= p2 and nums[i] == 2:
                Solution.swap(nums, i, p2)
                p2 -= 1
            if nums[i] == 0:
                Solution.swap(nums, i, p0)
                p0 += 1     
            i += 1    

执行用时：24 ms, 在所有 Python3 提交中击败了99.56%的用户
内存消耗：15 MB, 在所有 Python3 提交中击败了29.81%的用户