leetcode Java：75. 颜色分类

最新推荐文章于 2022-03-06 09:00:58 发布

原创最新推荐文章于 2022-03-06 09:00:58 发布 · 158 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode

leetcode-java 同时被 2 个专栏收录

90 篇文章

订阅专栏

双指针

23 篇文章

订阅专栏

该篇博客介绍了一种使用双指针算法解决数组中颜色分类的问题，具体是将0、1、2三个颜色的元素分别归位。通过一个指针从左向右移动，将0元素前移，另一个指针从右向左移动，将2元素后移，过程中保持中间位置为1元素。这种方法避免了使用额外空间，提高了效率。

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

题目：75. 颜色分类

思路：双指针。

一个指针指向0应该在的位置，一个指针指向2应该在的位置。遍历数组，把0交换到前面，把2交换到后面。

代码：

class Solution {
    public void sortColors(int[] nums) {
        int n = nums.length;
        int cur = 0, left = 0, right = n - 1;
        while (cur <= right) {
            if (nums[cur] == 0) {
                int temp = nums[cur];
                nums[cur] = nums[left];
                nums[left] = temp;
                cur ++;
                left ++;
            }
            else if (nums[cur] == 2) {
                int temp = nums[cur];
                nums[cur] = nums[right];
                nums[right] = temp;
                right --;
            }
            else {
                cur ++;
            }
        }
    }
}