算法题：Leetcode-75-sortColors 颜色分类（python）

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

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

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

 

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

"""

def sortColors(nums) -> None:
    """
    Do not return anything, modify nums in-place instead.
    """
    count = [0, 0, 0]
    for i in nums:
        count[i] += 1
        print(i,count[i])
    print(count)
    nums.clear()
    for i in range(3):
        for j in range(count[i]):
            nums.append(i)
                
                
def sortColors02(nums):
    zero = -1
    two = len(nums)
    i = 0
    while i < two:
        if nums[i] ==1:
            i += 1
        elif nums[i] == 0:
            zero += 1
            nums[zero],nums[i] = nums[i],nums[zero]
            i += 1
        else:
            two -= 1
            nums[two],nums[i] = nums[i],nums[two]
nums = [1,2,2,0]
sortColors02(nums)
print(nums)