【LeetCode刷题笔记】75.颜色分类

最新推荐文章于 2022-05-14 20:16:10 发布

C-Liueasymoney

最新推荐文章于 2022-05-14 20:16:10 发布

阅读量139

点赞数

分类专栏： LeetCode 双指针文章标签：指针数据结构快速排序 leetcode 算法

本文链接：https://blog.csdn.net/weixin_42050795/article/details/118049745

版权

LeetCode 同时被 2 个专栏收录

46 篇文章 0 订阅

订阅专栏

双指针

5 篇文章 0 订阅

订阅专栏

本文介绍了LeetCode第75题的解决方案，即如何对包含红、白、蓝三种颜色的数组进行排序，使得同色元素相邻且按红色、白色、蓝色顺序排列。采用双指针策略，通过不断调整数组元素，最终实现颜色分类。

摘要由CSDN通过智能技术生成

题目：给定一个包含红色、白色和蓝色，一共 n 个元素的数组，原地对它们进行排序，使得相同颜色的元素相邻，并按照红色、白色、蓝色顺序排列。
此题中，我们使用整数 0、 1 和 2 分别表示红色、白色和蓝色。

示例：输入：nums = [2,0,2,1,1,0]
输出：[0,0,1,1,2,2]

双指针
由于只要排序三个数字0、1、2，那么我们最终是要把0放在最左边，2放在最右边（假设0、2都存在）
那么就设置两个指针分别指向数组的首位和末尾，这两个指针的作用是用来当遍历到数组中元素等于0或者2的时候就分别与左指针和右指针调换。
其中要注意的是，调换结束后遍历到的当前元素仍有可能是2或者0，那么就需要把指针i回退一位再对第i位进行调换。
并且要对边界条件进行考虑，当i遍历到比右指针大的时候就应该停止了，因为此时i的位置应该要大于等于右指针的位置了，无需交换。

    public void sortColors(int[] nums){
        int left = 0;
        int right = nums.length - 1;
        for (int i = 0; i <= right; i++){
//            这里不用考虑指针回退，因为left指针在左边，已经进行过调整不会出现交换过来一个2的情况
            if (nums[i] == 0) {
                nums[i] = nums[left];
                nums[left] = 0;
                left++;
            }

            if (nums[i] == 2){
                nums[i] = nums[right];
                nums[right] = 2;
                right--;
//                指针i回退！！因为调换过来如果不是1而是0或者2还需要对第i位进行调整
                if (nums[i] != 1)
                    --i;
            }
        }
    }