Lintcode: 颜色分类

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

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

您在真实的面试中是否遇到过这个题？ Yes
样例
注意
不能使用代码库中的排序函数来解决这个问题

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

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

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

方法一

按照lintcode提示的方法，使用一个map，迭代数组，计算0，1，2出现的次数，然后依次使用次数重新给nums数组赋值，复杂度为O(n^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
        map<int, size_t> word_count;
        for(int i=0; i!=nums.size(); i++){
            ++word_count[nums[i]];
        }
        int i = 0;
        for(auto b=word_count.begin(); b!=word_count.end(); b++){
            for(int j=0; jsecond; j++){
                nums[i++] = b->first;
            }
        }
    }
};

方法二

另一种方法是只遍历一次，把nums分成三部分，把0放在nums的前部分，2放在后部分，1自然就在中间部分，复杂度为O(n)。
参考

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 l = 0;
        int r = nums.size()-1;
        for(int i=0;iint)nums.size(),r+1);)
        {
            if(nums[i]==0)
            {
                if(i==l)
                    i++;
                else
                    swap(nums[i],nums[l]);
                l++;
            }
            else if(nums[i]==2)
            {
                swap(nums[i],nums[r]);
                r--;
            }
            else
                i++;
        }
    }
};