lintcode-排颜色II-143

给定一个有n个对象（包括k种不同的颜色，并按照1到k进行编号）的数组，将对象进行分类使相同颜色的对象相邻，并按照1,2，...k的顺序进行排序。

样例

给出colors=[3, 2, 2, 1, 4]，k=4, 你的代码应该在原地操作使得数组变成[1, 2, 2, 3, 4]

注意

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

挑战

一个相当直接的解决方案是使用计数排序扫描2遍的算法。这样你会花费O(k)的额外空间。你否能在不使用额外空间的情况下完成？

解题思路：运用快排中数组划分的思想

class Solution{
public:
      
    int solve(vector<int> &v,int begin,int end,int color){  
        int slow=begin-1,fast=begin;  
        while(fast<end){  
            if(v[fast]==color)  
                swap(v[fast],v[++slow]);  
            ++fast;      
        }  
        return slow+1;  
    }    
    void sortColors2(vector<int> &colors, int k) {
        int begin=0,end=colors.size();
        
        for(int i=1;i<=k;++i){
            begin=solve(colors,begin,end,i);
        }
    }
};