给定一个包含红色、白色和蓝色,一共 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