排颜色 II -LintCode

给定一个有n个对象（包括k种不同的颜色，并按照1到k进行编号）的数组，将对象进行分类使相同颜色的对象相邻，并按照1,2，…k的顺序进行排序。
注意事项
You are not suppose to use the library’s sort function for this problem.
k <= n
样例
给出colors=[3, 2, 2, 1, 4]，k=4, 你的代码应该在原地操作使得数组变成[1, 2, 2, 3, 4]
挑战
一个相当直接的解决方案是使用计数排序扫描2遍的算法。这样你会花费O(k)的额外空间。你否能在不使用额外空间的情况下完成？

#ifndef C143_H
#define C143_H
#include<iostream>
#include<vector>
using namespace std;
class Solution{
public:
    /**
    * @param colors: A list of integer
    * @param k: An integer
    * @return: nothing
    */
    void sortColors2(vector<int> &colors, int k) {
        // write your code here
        int len = colors.size();
        if (len <= 0)
            return;
        int i = 0;
        while (i < len)
        {
            int pos = colors[i] - 1;
            if (colors[i] <= 0)
                ++i;
            else
            {
                if (colors[pos] <= 0)
                {
                    colors[pos]--;
                    colors[i] = 0;
                    ++i;
                }
                else
                {
                    swap(colors[i], colors[pos]);
                    colors[pos] = -1;
                }
            }
        }
        int j = len - 1;
        while (k > 0)
        {
            for (int m = 0; m > colors[k - 1]; --m)
                colors[j--] = k;
            --k;
        }
    }
};
#endif