荷兰国旗问题引出的三色排序

现在有若干个红、白、蓝三种颜色的球随机排列成一条直线。现在我们的任务是把这些球按照红、白、蓝排序，就类似于荷兰国旗一样。

这个问题我们可以将这个问题视为一个数组排序问题，这个数组分为前部，中部和后部三个部分，每一个元素（红白蓝分别对应0、1、2）必属于其中之一。

由于红、白、蓝三色小球数量并不一定相同，所以这个三个区域不一定是等分的，也就是说如果我们将整个区域放在[0,1]的区域里，由于三色小球之间数量的比不同（此处假设1:2:2），可能前部为[0,0.2)，中部为[0.2,0.6)，后部为[0.6,1]。我们的思路如下：将前部和后部各排在数组的前边和后边，中部自然就排好了。

具体方法如下：

设置两个标志位begin和end分别指向这个数组的开始和末尾，然后用一个标志位i从头开始进行遍历：
1）若遍历到的位置为0，则说明它一定属于前部，于是就和begin位置进行交换，然后i向前进，begin也向前进（表示前边的已经都排好了）。
2）若遍历到的位置为1，则说明它一定属于中部，根据总思路，中部的我们都不动，然后i向前进。

3）若遍历到的位置为2，则说明它一定属于后部，于是就和end位置进行交换，由于交换完毕后i指向的可能是属于前部的，若此时i前进则会导致该位置不能被交换到前部，所以此时i不前进。而同1），end向前进一。

具体代码如下：

class Solution {
public:
    void sortColors(int A[], int n) {
        int begin = 0;
        int end = n-1;
        int i = 0;
        while(i<end+1)
        {
            if(A[i]==0)
            {
                swap(A[i], A[begin]);
                begin++;
                i++;
            }
            else if(A[i]==2)
            {
                swap(A[i], A[end]);
                end--;
                //i++;
            }
            else
                i++;
        }
    }
};