思路:
设置三个标记指针,pos0,pos2,cur
令pos0从前往后遍历,指向第一个非0的位置,pos2从后往前遍历,指向第一个非2的位置
然后cur从pos0开始往后遍历:
遇到0就和pos0交换,pos0++;
遇到1什么也不做;
遇到2就和pos2交换,pos2向前滑动到下一个非2的位置,交换后还要重新检查cur的值;
直到cur与pos2相遇。
一次遍历,复杂度是O(n),因为每次操作都使得数组更为有序,不像快排需要重复比较,所以比应用快排的方法效率高一些。
具体代码实现:
int main(){
int arr[14] = {0,1,2,1,2,0,2,0,1,2,2,2,1,2};
int pos0=0, pos2=sizeof(arr)/4 -1, pcur=0;
while(0 == arr[pos0])
++pos0;
while(2 == arr[pos2])
--pos2;
pcur = pos0;
while(pcur <= pos2){
if(0 == arr[pcur]){
swap(arr[pcur], arr[pos0]);
++pos0;
}
else if(2 == arr[pcur]){
swap(arr[pcur], arr[pos2]);
if(0 == arr[pcur]){//若交换之后,pcur当前指向的元素为0,则继续将pcur指向的元素和pos0指向的元素进行交换
swap(arr[pcur], arr[pos0]);
++pos0;//交换之后,将pos0向前移动一位
}
--pos2;//pos2向后移动一位
while(arr[pos2] == 2)//若移动之后指向的元素还是2,则继续向前移动,直到指向第一个非2的元素
--pos2;
}
++pcur;//将pcur向前移动
}
//将原数组打印出来
for(int i =0;i < sizeof(arr)/4;++i)
cout<<arr[i]<<" ";
cout<<endl;
return 0;
}
结果如图所示: