三色排序练习题

有一个只由0，1，2三种元素构成的整数数组，请使用交换、原地排序而不是使用计数进行排序。
给定一个只含0，1，2的整数数组A及它的大小，请返回排序后的数组。保证数组大小小于等于500。
解题思路：
思路1:

直接给整个数组排序，时间复杂度O(nlogn)

思路2:

计数排序方法，使用一个类似hash的数组纪录每种颜色的个数，然后进行排序，但是需要遍历原数组两遍

思路3:

使用三个指针，p1表示红色和白色的分界线，p2表示白色和蓝色的分界线，i表示当前元素

即0~p1-1是红色的，p1~i-1表示白色的，p2+1~n-1表示蓝色的

1）如果当前元素是红色的，则和p1所指向的元素进行交换，由于交换以后i所指的颜色是白色的，则i直接遍历下一个元素

2）如果当前元素是蓝色的，则和p2所指向的元素进行交换，由于交换以后i所指的颜色可能是白色的，也可能是红色的，因此需要回退i

 public int[] sortThreeColor(int[] A, int n){
        // write code here
        int left = -1;
        int right = A.length;
        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;
    }
    public void swap(int[] A,int i, int j){
        int temp = A[i];
        A[i] = A[j];
        A[j] = temp;
    }