LeetCode 143.颜色分类Ⅱ

最新推荐文章于 2021-10-13 09:56:46 发布

原创最新推荐文章于 2021-10-13 09:56:46 发布 · 149 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#lintcode #java #快速排序 #归并排序

LintCode 专栏收录该内容

51 篇文章

订阅专栏

本文探讨了一种问题：给定n个具有k种不同颜色的对象，如何在保持颜色相同对象相邻且按颜色编号排序的情况下，实现高效的算法。通过分析，确定了时间复杂度为O(nlogk)，涉及快速排序和归并排序的结合。作者提供了Java代码示例，展示了如何使用rainbowSort方法解决此问题。

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

通过时间复杂度推测用什么算法！
k = 1, O(1)
k = 2, O(n)
k = 3, O(n)
…
k = n, O(nlogn) //每个元素都不相同快速排序
所以本题的算法的时间复杂度应为 nlogk
故应该是将k个颜色分治归并每一层都要对n个元素进行排序处理
快速排序+归并排序

public class Solution {
    /**
     * @param colors: A list of integer
     * @param k: An integer
     * @return: nothing
     */
    public void sortColors2(int[] colors, int k) {
        if(colors == null || colors.length == 0)
            return;
        rainbowSort(colors, 0, colors.length- 1, 1, k);
    }
    public void rainbowSort(int[] colors, int left, int right, int colorFrom, int colorTo){
        if(colorFrom == colorTo)
            return;
        if(left >= right) 
            return;
        int colorMid = (colorTo + colorFrom) / 2;
        int l = left, r = right;
        while(l <= r){
            while(l <= r && colors[l] <= colorMid){
                l++;
            }
            while(l <= r && colors[r] > colorMid){
                r--;
            }
            if(l <= r){
                int temp = colors[l];
                colors[l] = colors[r];
                colors[r] = temp;
                l++;
                r--;
            }
        }
        rainbowSort(colors, left, r, colorFrom, colorMid);
        rainbowSort(colors, l, right, colorMid + 1, colorTo);
        
    }
}