LeetCode 75：颜色分类

最新推荐文章于 2022-04-05 18:39:59 发布

香农派我最爱

最新推荐文章于 2022-04-05 18:39:59 发布

阅读量283

点赞数

分类专栏： LeetCode 文章标签： leetcode 算法排序算法

本文链接：https://blog.csdn.net/weixin_43161461/article/details/103739715

版权

LeetCode 专栏收录该内容

40 篇文章 0 订阅

订阅专栏

LeetCode 75：颜色分类

题目描述

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

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

注意:
不能使用代码库中的排序函数来解决这道题。

示例:

输入: [2,0,2,1,1,0]
输出: [0,0,1,1,2,2]
进阶：

一个直观的解决方案是使用计数排序的两趟扫描算法。
首先，迭代计算出0、1 和 2 元素的个数，然后按照0、1、2的排序，重写当前数组。
你能想出一个仅使用常数空间的一趟扫描算法吗？

解题思路

本题出自，荷兰三色旗问题，题目需要借助三个指针，p0，curr，p2，分别针对元素0，1，2。p0和curr初始位置数组开头，p2的初始位置是数组末尾。思想如下：在这里插入图片描述

代码实现

去博客设置页面，选择一款你喜欢的代码片高亮样式，下面展示同样高亮的 代码片.

public class sortColor {
    //颜色分类，荷兰三色旗问题
    public void sortColors(int[] nums){ //acc:0ms 100%

        int p0=0,curr=0;//p0是0的初始化位置，curr是1的初始化位置
        int p2=nums.length-1;//p2是2的初始化位置

        int tmp;//元素交换的临时变量
        while (curr<=p2) {
            if (nums[curr] == 0) { //当前指针所指元素和0一样，应该和p0指针互换元素,
                //并且p0和curr指针均向后移
                tmp = nums[p0];
                nums[p0++] = nums[curr];
                nums[curr++] = tmp;
            } else if (nums[curr] == 2) {//nums[curr]==2,和p2互换元素，p2前移
                tmp = nums[curr];
                nums[curr] = nums[p2];
                nums[p2--] = tmp;
            } else {
                curr++;//nums[curr]==1

            }


        }
    }

    public static void main(String[] args) {
        sortColor test=new sortColor();
        int[] nums={2,0,2,1,1,0};
        test.sortColors(nums);
        System.out.println(nums[0]+","+nums[1]);

    }
}