LintCode : 排颜色 II

LintCode : 排颜色 II

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

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

方法一

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
        map<int, int> imap;
        for(int i=0; i<colors.size(); i++){
            imap[colors[i]] += 1;
        }
        int m = 0;
        for(int i=0; i<imap.size(); i++){
            for(int j=0; j<imap[i]; j++){
                colors[m++] = i;
            }
        }
    }
};

方法二

题目要求不使用额外的数组，可以利用上面的思想，只不过要用原来的数组a来统计每个颜色出现的频率。由于原来的颜色肯定是正数1-k，所以我们可以用负数比如a[i]=-k,表示第i种颜色在原来的数组里面出现了k次。这道题原数组还要重复利用作为bucket数组，那么我们应该怎么办呢？首先for循环遍历一遍原来的数组，如果扫到a[i]，首先检查a[a[i]]是否为正数，如果是把a[a[i]]移动a[i]存放起来，然后把a[a[i]]记为-1(表示该位置是一个计数器，计1)。 如果a[a[i]]是负数，那么说明这一个地方曾经已经计数了，那么把a[a[i]]计数减一，并把color[i] 设置为0 （表示此处已经计算过），然后重复向下遍历下一个数，这样遍历原数组所有的元素过后，数组a里面实际上存储的每种颜色的计数，然后我们倒着再输出每种颜色就可以得到我们排序后的数组。