Sort Colors II

最新推荐文章于 2017-04-07 17:29:07 发布

原创最新推荐文章于 2017-04-07 17:29:07 发布 · 413 阅读

0 ·

CC 4.0 BY-SA版权

strStr 同时被 3 个专栏收录

360 篇文章

订阅专栏

68 篇文章

订阅专栏

开发智力

67 篇文章

订阅专栏

本文介绍了一种针对颜色数组的颜色排序算法实现。该算法每次只对数组中的两种颜色进行排序，同时保持其他元素不变，并通过迭代逐步完成整个数组的排序。文章提供了一个具体的Java代码示例。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

参考点击打开链接

每次只对两个颜色min 和max排序，其他的都不变，于是中间再用I的方法即可求解。

class Solution {
    /**
     * @param colors: A list of integer
     * @param k: An integer
     * @return: nothing
     */
    public void sortColors2(int[] colors, int k) {
        // write your code here
        
        int start = 0, end = colors.length - 1, count = 0;
        while (count < k) {
            int min = Integer.MAX_VALUE;
            int max = Integer.MIN_VALUE;
            for (int i = start; i <= end; i++) {
                min = Math.min(min, colors[i]);
                max = Math.max(max, colors[i]);
            }
            
            int left = start, cur = start, right = end;
            while (cur <= right) {
                if (colors[cur] == min) {
                    swap(colors, left, cur);
                    left++;
                    cur++;
                } else if (colors[cur] > min && colors[cur] < max) {
                    cur++;
                } else {
                    swap(colors, right, cur);
                    right--;
                }
            }
            count = count + 2;
            start = left;
            end = right;
        }
    }
    
    private void swap(int[] colors, int right, int cur) {
        int temp = colors[right];
        colors[right] = colors[cur];
        colors[cur] = temp;
    }
}