那些年我们刷过的算法题（排序）---有一个只由0，1，2三种元素构成的整数数组，请使用交换、原地排序而不是使用计数进行排序

有一个只由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;
}
}
}