143.Sort Colors II-排颜色 II（中等题）

排颜色 II

1. 题目

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

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

2. 样例

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

3. 挑战

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

4. 题解

   由于要求不适用额外空间，我们这里用colors数组来存每种颜色出现的次数，colors[k - 1]表示颜色k出现的次数(用负数表示以示区分)。首先遍历一遍计算出每种颜色出现的次数，再从尾部开始填充颜色。

class Solution {
    /**
     * @param colors: A list of integer
     * @param k: An integer
     * @return: nothing
     */
    public void sortColors2(int[] colors, int k) {
        for (int i=0;i<colors.length;)
        {
            if (colors[i] <= 0)
            {
                i++;
                continue;
            }
            else if (colors[colors[i]-1] >= 0)
            {
                int tmp = colors[i]-1;
                swap(colors, i, colors[i] - 1);  
                colors[tmp] = -1;  
            }
            else
            {
                colors[colors[i]-1]--;
                colors[i] = 0;
                i++;
            }
        }
        int index = colors.length - 1;  
        int curK = k;  
        while(index >= 0) 
        {  
            while(colors[curK - 1] < 0) 
            {  
                colors[index--] = curK;  
                if(index + 1 != curK - 1)  
                {
                    colors[curK - 1]++; 
                }
            }  
            curK--; 
        }
    }

    private void swap(int[] colors, int a, int b) 
    {  
        int tmp = colors[a];  
        colors[a] = colors[b];  
        colors[b] = tmp;  
    }  
}

Last Update 2016.10.17