Leetcode.75 颜色分类

最新推荐文章于 2026-01-06 22:04:30 发布

原创最新推荐文章于 2026-01-06 22:04:30 发布 · 200 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#算法 #leetcode

Leetcode/剑指offer-【数组篇】专栏收录该内容

72 篇文章

订阅专栏

文章描述了一种原地排序算法，用于将给定的整数数组（0,1,2表示红、白、蓝）按照红、白、蓝顺序排列，不使用内置sort函数。通过双指针技巧实现相同颜色元素相邻，且颜色顺序正确。

给定一个包含红色、白色和蓝色、共 n 个元素的数组 nums ，原地对它们进行排序，使得相同颜色的元素相邻，并按照红色、白色、蓝色顺序排列。

我们使用整数 0、 1 和 2 分别表示红色、白色和蓝色。

必须在不使用库内置的 sort 函数的情况下解决这个问题。

class Solution {
    public void sortColors(int[] nums) {
        int n0 = 0, n1 = 0;
        //[2,0,2,1,1,0],[1,0,2,2,2]
        for(int i = 0; i < nums.length; i++){
            int num = nums[i];
            nums[i] = 2;
            if(num < 2){
                nums[n1++] = 1;
            }
            if(num < 1){
                nums[n0++] = 0;
            }
        }

    }
}

解析：

我来解释一下。首先，所有数都≤2，那么索性把所有数组置为2，然后遇到所有≤1的，就再全部置为1，，覆盖了错误的2，这时候所有2的位置已经正确，最后所有≤0的全部置为0，也就覆盖了一些错误的1，这时候，0和1的位置都正确。最重要的就是需要两个指针指向下一个1或0的位置

遇到0的情况，先变为1，再变为0，所以现在后面变为1，在前面覆盖0，不会出现0覆盖1的问题