那些年我们刷过的算法题（排序）---有一个只由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;
        } 
    }
}