Leetcode - 75. 颜色分类

给定一个包含红色、白色和蓝色，一共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
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

解1 2021/9/10 O(n)

from collections import defaultdict

def sortColors(nums: list) -> None:
    """
    Do not return anything, modify nums in-place instead.
    """
    '''
    这题遍历，数目，存下来，重新赋值就行了
    '''
    #print(id(nums))
    d=defaultdict(int)
    for x in nums:
        d[x]+=1
    # d的key只有0,1,2
    # 当重新赋值时，id就变了，里外打出来的并非一个值
    #nums=[0]*d[0]+[1]*d[1]+[2]*d[2]
    #print(id(nums))
    i=0
    for x in range(0,3):
        for y in range(0,d[x]):
            nums[i]=x
            i+=1
    #print(nums)

if __name__ == '__main__':
    nums = [2, 0, 2, 1, 1, 0]
    sortColors(nums)
    nums = [2, 0, 1]
    sortColors(nums)
    nums = [0]
    sortColors(nums)
    nums = [1]
    sortColors(nums)