有一个只由0,1,2三种元素构成的整数数组,请使用交换、原地排序而不是使用计数进行排序
只有三个数字,只能用交换,原地排序,其实交换次数最少,我为什么马上就想到了类似于快排的数组划分过程,
思路:从第一个元素开始遍历,如果是1则位置不变,如果是0与前面第一个元素交换left++,如果是2与最后元素交换right++,
遍历过程用i做标记,与元素交换的位置用left和right做标记,知道i大于right的值为止。
这样:时间复杂度为O(n),空间复杂度为O(1)
代码:
public class ThreeColor {
public int[] sortThreeColor(int[] A, int n) {
// write code here
int left = 0;
int right = n-1;
int i = 0;
while(i <= right){
if(A[i] == 0){
swap(A,i,left++);
i++;
}else if(A[i] == 2){
swap(A,i,right--);
}else
i++;
}
return A;
}
private void swap(int[] A,int a,int b){
if(a != b){
int temp = A[a];
A[a] = A[b];
A[b] = temp;
}
}
}