【刷题之路】三色排序问题

有一个只由0，1，2三种元素构成的整数数组，请使用交换、原地排序而不是使用计数进行排序。

由于不能使用计数排序，又只有三种元素，可以考虑分块处理。

在数组头开辟一个程度为0的区域left，在尾部开一个长度为0的区域right

开始遍历数组，如果元素为0，则将该元素移入left中，并将left区域长度+1，同理，如果元素为2，则将该元素移入队尾。这里所说的移入，实际上是交换。同时需要注意，与0交换的元素我们已经遍历过了，所以可以确定与0交换的不是0，但是与2交换的元素处在队尾，我们并不知道这个元素是什么，所以我们需要再一次遍历该位置上元素，决定其去留。代码如下

class ThreeColor {
public:
    vector<int> sortThreeColor(vector<int> A, int n) {
        // write code here
        int temp,i=0,left=0,right=n-1;  //left与right分别为0与2区域的边界
        while(i<n&&i<=right){ //这里注意如果遍历到2区域的边界，则遍历可以停止了，因为后面可以确定都是2了
            if(A[i]==0){
                swap(A[i],A[left]);  //当元素为0时，将0放入队头同时扩大0区域的size，并向后遍历
                ++left;
                ++i;
            }
            else if(A[i]==2){
                swap(A[i],A[right]); //当元素为2是，将2放入队尾同时扩大2区域的size，这时候不能向后遍历而是重复遍历这个位置的值，直到这个位置的值不是2时，再向后
                --right; 
            }
            else{  //当元素为1时，向后遍历
                ++i;
            }
        }
        return A;
    }
};