lintcode--颜色分类

给定一个包含红，白，蓝且长度为 n 的数组，将数组元素进行分类使相同颜色的元素相邻，并按照红、白、蓝的顺序进行排序。

我们可以使用整数 0，1 和 2 分别代表红，白，蓝。

 注意事项

不能使用代码库中的排序函数来解决这个问题。
排序需要在原数组中进行。

您在真实的面试中是否遇到过这个题？ 

Yes

样例

给你数组 [1, 0, 1, 2], 需要将该数组原地排序为[0, 1, 1, 2]。

採用高速排序partition的思想，用两个指针将三种颜色间隔开。


以下引用此处内容 July 编程艺术系列中荷兰国旗问题。
一个前指针begin，一个中指针current，一个后指针end，current指针遍历整个数组序列，当
current指针所指元素为0时，与begin指针所指的元素交换，而后current++，begin++ 。
current指针所指元素为1时，不做什么交换（即球不动），而后current++ 。
current指针所指元素为2时，与end指针所指的元素交换，而后，current指针不动，end-- 。




为什么上述第3点中，current指针所指元素为2时，与end指针所指元素交换之后，current指针不能动呢？由于第三步中current指针所指元素与end指针所指元素交换之前，假设end指针之前指的元素是0，那么与current指针所指元素交换之后。current指针此刻所指的元素是0，此时。current指针能动么？不能动，由于如上述第1点所述。假设current指针所指的元素是0，还得与begin指针所指的元素交换。


     */
public class Solution {
    
    public void sortColors(int[] A){
        int begin = 0;
        int end =A.length-1;
        int cur = 0;
        while(cur<=end){
            //current指针所指元素为0时，与begin指针所指的元素交换，而后current++，begin++ 。
            if(A[cur] ==0){
                swap(A,begin,cur);
                begin++;
                cur++;
            //current指针所指元素为1时，不做什么交换（即球不动），而后current++ 
            }else if(A[cur] == 1){
                cur++;
            }else {
            //current指针所指元素为2时，与end指针所指的元素交换，而后，current指针不动，end--
                swap(A,end,cur);
                end--;
            }
        }
    }
    private void swap(int[]A,int cur,int j){
        int temp =A[cur];
        A[cur] = A[j];
        A[j] = temp;
    }


}