leet75. 分类颜色

题目：

给定一个包含红色、白色和蓝色，且含有 n 个元素的数组，对它们进行排序，使得相同颜色的元素相邻，颜色顺序为红色、白色、蓝色。

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

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

点击显示进阶问题.

进阶：
一个相当直观的解决方案是使用计数排序的 two-pass 算法。

首先，迭代计算出0，1 和 2 元素的个数，然后重写当前数组。

你能想出一个仅使用恒定空间的 one-pass 算法吗？

分析：

1. 输出的结果形式为[0]*[1]*[2]*
2. 统计0和2的个数rNums和bNums，并在遍历过程中，将已遍历元素修改为1
3. 遍历结束后，将前rNums个元素改为0，末尾bNums个元素改为2
   

代码：

class Solution(object):
    def sortColors(self, nums):
        """
        :type nums: List[int]
        :rtype: void Do not return anything, modify nums in-place instead.
        """
        if len(nums) > 1:
            rNums = 0
            bNums = 0
            for i,x in enumerate(nums):
                if x == 0: rNums += 1
                if x == 2: bNums += 1
                nums[i] = 1
            if rNums:
                nums[:rNums] = [0] * rNums
            if bNums:
                nums[-bNums:] = [2] * bNums

思考：

1. 采用原址排序，该代码时间复杂度O(n)，空间复杂度O(1)