LeetCode--Sort Colors（颜色排序）Python

题目：

给定一个长度为n的数组，数组中包含三种颜色的球。将相同颜色的球连接排列。返回重新排列后的数组。其中0、1、2分别代表一个颜色

解题思路：

统计数组中0、1、2分别出现的次数。根据次数重新对数组进行赋值。

代码（Python）：

class Solution(object):
    def sortColors(self, nums):
        """
        :type nums: List[int]
        :rtype: void Do not return anything, modify nums in-place instead.
        """
        count0 = 0
        count1 = 0
        count2 = 0
        for i in nums:
            if i==0:
                count0+=1
            elif i==1:
                count1+=1
            else:
                count2+=1
        
        output = []
        for i in range(count0):
            nums[i] = 0
        for i in range(count1):
            nums[i+count0] = 1
        for i in range(count2):
            nums[i+count0+count1] = 2