[Java·算法·中等]LeetCode75. 颜色分类

最新推荐文章于 2024-10-01 23:05:47 发布

九尾狐的yi巴

最新推荐文章于 2024-10-01 23:05:47 发布

阅读量65

点赞数

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

本文链接：https://blog.csdn.net/jiuweihu521/article/details/129758043

版权

算法专栏收录该内容

50 篇文章 1 订阅

订阅专栏

文章讨论了一种对包含红、白、蓝三种颜色元素的数组进行排序的方法，要求相同颜色的元素相邻且按红色、白色、蓝色顺序排列。提供了两种解决方案：一种是使用双指针，另一种是采用计数排序。两种方法均避免了使用内置排序函数，实现了原地排序。

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

👉️ 力扣原文

题目

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

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

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

示例

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

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

分析思路1

单指针:
我们可以考虑对数组进行两次遍历。在第一次遍历中，我们将数组中所有的 000 交换到数组的头部。在第二次遍历中，我们将数组中所有的 111 交换到头部的 000 之后。此时，所有的 222 都出现在数组的尾部，这样我们就完成了排序。

题解1

class Solution {
    public void sortColors(int[] nums) {
        int n = nums.length;
        int ptr = 0;
        for (int i = 0; i < n; ++i) {
            if (nums[i] == 0) {
                int temp = nums[i];
                nums[i] = nums[ptr];
                nums[ptr] = temp;
                ++ptr;
            }
        }
        for (int i = ptr; i < n; ++i) {
            if (nums[i] == 1) {
                int temp = nums[i];
                nums[i] = nums[ptr];
                nums[ptr] = temp;
                ++ptr;
            }
        }
    }
}

执行结果
在这里插入图片描述

分析思路2

计数排序：
统计出数组中每个元素出现的个数，然后从小到大依次遍历计数数组，再将这个元素放回到原数组中对应的位置。

题解2

class Solution {
    public void sortColors(int[] nums) {
        int[] shorts = new int[3];
        for(int num:nums){
            shorts[num]++;
        }
        int j=0;
        for(int i=0; i<shorts.length; i++){
            while(shorts[i]-->0){
                nums[j++]=i;
            }
        }
    }
}