LeetCode 75*. 颜色分类（Python）

 

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

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

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

示例:

输入: [2,0,2,1,1,0]
输出: [0,0,1,1,2,2]

仅使用常数空间的一趟扫描算法

 思路：参考三路快排，定义两个指针zero和two，使得nums中从0到zero之间的数全为0，从two到n-1位置的数全为2，zero到i之间的数全为1

定义循环索引i

如果nums[i] == 0则将nums[zero+1]与nums[i]互换，zero指针向后移动一位，i也向后移动一位

如果nums[i] == 2则将 nums[two-1]与nums[i]互换，two指针向前移动一位

如果nums[i] == 1则i++

 

class Solution(object):
    def sortColors(self, nums):
        """
        :type nums: List[int]
        :rtype: None Do not return anything, modify nums in-place instead.
        """
        zero = -1   # nums[0 to zero] == 0
        two = len(nums) # nums[two to n-1] == 2
        i = 0
        while i < two:
            if nums[i] == 1:
                i = i + 1
            elif nums[i] == 2:
                two = two - 1
                nums[i], nums[two] = nums[two], nums[i]     
            elif nums[i] == 0:
                zero = zero + 1
                nums[zero], nums[i] = nums[i], nums[zero]   
                i = i + 1   # i要随着zero动，i不能比他小
        
        return nums