题目:
给定一个包含红色、白色和蓝色,一共 n 个元素的数组,原地对它们进行排序,使得相同颜色的元素相邻,并按照红色、白色、蓝色顺序排列。
此题中,我们使用整数 0、 1 和 2 分别表示红色、白色和蓝色。
注意:
不能使用代码库中的排序函数来解决这道题。
来源:力扣(LeetCode)
解:
这道题禁止使用排序函数,那么也意味着可以有别的方法。
因为颜色数字只有三个,已经确定,所以第一种方法,统计数字个数,直接重构数组,两趟O(n)完成。
这里介绍一种复杂一点的双指针算法。
我们可以想出,最后的数组肯定是3个分类,两个中间边界,那么我们就可以尝试用中间边界来构造分区。如果采用确定0和2的边界,我们就从两边向中间移动扫描,0往前面放,2往后面放,但是这里有一个bug,如果0指针遇到了1,2指针也遇到了1怎么放?如果划过又会导致分区错误,不划过就卡在这里了,所以这种方法不采用。
我们构造0和1分区,从前往后扫描,如果0就放在0分区,1就放在1分区,然后被取代的值换到扫描的i位置。2不操作(即在最后的2分区)。那么我们就需要两个分区指针,一个代表0的末尾,一个代表1分区的末尾。但是这里又有另外一个问题,0分区是动态的,势必会冲突掉1分区的头部元素,所以我们需要对每次0分区的调换操作后检查被换出的是不是1,如果是,重新调回1分区末尾,这样就确保1分区的正确性。1分区的调换不需要操作,因为扫描到第i个元素,末尾是...1(即i刚好是1的边界,相当于换自己)或...2。
class Solution {
public void sortColors(int[] nums) {
int zero=0,one=0;
int t;
for(int i=0;i<nums.length;i++){
if(nums[i]==0){
t=nums[zero];
nums[zero++]=nums[i];
nums[i]=t;
}
if(nums[i]==1){
if(one<zero){
//初始化问题
one=zero;
}
t=nums[one];
nums[one++]=nums[i];
nums[i]=t;
}
}
}
}