颜色分类

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

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

样例

注意

不能使用代码库中的排序函数来解决这个问题

说明

一个相当直接的解决方案是使用计数排序扫描2遍的算法。

首先，迭代数组计算0,1,2出现的次数，然后依次用0,1,2出现的次数去覆盖数组。

你否能想出一个仅使用常数级额外空间复杂度且只扫描遍历一遍数组的算法？

class Solution{
public:
    /**
     * @param nums: A list of integer which is 0, 1 or 2 
     * @return: nothing
     */    
    void sortColors(vector<int> &nums) {
        // write your code here
        int n = nums.size();
        int left = 0;
        int right = n-1;
        int p = 0;
        while (p < n && left < right)
        {
            if (nums[p] == 0)
            {
                while (nums[left] == 0)
                {
                    left++;
                }
                if (p > left)
                {
                    swap(nums[left], nums[p]);
                }
                else    
                {
                    p = left + 1;
                }
            }
            else if (nums[p] == 2)
            {
                while (nums[right] == 2)
                {
                    right--;
                }
                if (p < right)
                {
                    swap(nums[right], nums[p]);
                }
                else
                {
                    break;
                }
            }
            else
            {
                p++;
            }
        }
    }
};