有一个只由0,1,2三种元素构成的整数数组,请使用交换、原地排序而不是使用计数进行排序。
由于不能使用计数排序,又只有三种元素,可以考虑分块处理。
在数组头开辟一个程度为0的区域left,在尾部开一个长度为0的区域right
开始遍历数组,如果元素为0,则将该元素移入left中,并将left区域长度+1,同理,如果元素为2,则将该元素移入队尾。这里所说的移入,实际上是交换。同时需要注意,与0交换的元素我们已经遍历过了,所以可以确定与0交换的不是0,但是与2交换的元素处在队尾,我们并不知道这个元素是什么,所以我们需要再一次遍历该位置上元素,决定其去留。代码如下
class ThreeColor {
public:
vector<int> sortThreeColor(vector<int> A, int n) {
// write code here
int temp,i=0,left=0,right=n-1; //left与right分别为0与2区域的边界
while(i<n&&i<=right){ //这里注意如果遍历到2区域的边界,则遍历可以停止了,因为后面可以确定都是2了
if(A[i]==0){
swap(A[i],A[left]); //当元素为0时,将0放入队头同时扩大0区域的size,并向后遍历
++left;
++i;
}
else if(A[i]==2){
swap(A[i],A[right]); //当元素为2是,将2放入队尾同时扩大2区域的size,这时候不能向后遍历而是重复遍历这个位置的值,直到这个位置的值不是2时,再向后
--right;
}
else{ //当元素为1时,向后遍历
++i;
}
}
return A;
}
};