利用python 完成leetcode75 颜色分类

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

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

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

示例:

输入: [2,0,2,1,1,0]
输出: [0,0,1,1,2,2]
进阶：

一个直观的解决方案是使用计数排序的两趟扫描算法。
首先，迭代计算出0、1 和 2 元素的个数，然后按照0、1、2的排序，重写当前数组。
你能想出一个仅使用常数空间的一趟扫描算法吗？
思路
用i遍历数组
用a储存遇到0应该放到的位置，遇到后将nums[i]与nums[a]交换并a+1
用b储存遇到2应该放到的位置，遇到后将nums[i]与nums[b]交换，再寻找下一个值不为2的位置（为了防止num[b-1]=2。a不用，因为nums[a+1]不可能等于0，i是从前向后遍历的，nums[a]到nums[i]之间不会有0）
代码

def sortColors(self, nums):
        i=0
        a=0
        b=len(nums)-1
        while b>=0 and nums[b]==2 :b=b-1
        while a<len(nums)and nums[a]==0 :a=a+1
        i=a
        while i<=b:
            if nums[i]==0:
                nums[i],nums[a] = nums[a],nums[i]
                a+=1
            elif nums[i] == 2:
                nums[i],nums[b] = nums[b],nums[i]                
                while nums[b]==2 and b>=0:b=b-1
                i=i-1#防止换过来的是0，重新检查一下
            i+=1