leetcode 75.颜色分类

最新推荐文章于 2024-09-19 10:43:15 发布

づ淺帞づ

最新推荐文章于 2024-09-19 10:43:15 发布

阅读量112

点赞数 1

分类专栏： leetcode 文章标签：算法数据结构 leetcode 排序算法

本文链接：https://blog.csdn.net/weixin_43140586/article/details/118974549

版权

leetcode 专栏收录该内容

39 篇文章 1 订阅

订阅专栏

该博客介绍了一种解决排序含有红、白、蓝三种颜色元素数组的方法，使用了计数排序算法。在O(n)的时间复杂度和O(1)的空间复杂度下，实现了原地排序，确保相同颜色的元素相邻且按红、白、蓝顺序排列。文章提供了详细的解题思路和Java代码实现。

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

文章目录

1、题目

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

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

示例 1：

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

示例 2：

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

示例 3：

输入：nums = [0]
输出：[0]

示例 4：

输入：nums = [1]
输出：[1]

提示：

n == nums.length
1 <= n <= 300
nums[i] 为 0、1 或 2

进阶：

你可以不使用代码库中的排序函数来解决这道题吗？
你能想出一个仅使用常数空间的一趟扫描算法吗？

2、解题思路

（1）解决方案一：计数排序；（时间复杂度为O(n)；空间复杂度为O(1) ）

-<1> 创建一个数组A，长度为2；

-<2> 遍历原始数组，遇到对应的元素就在A对应下标的值加1

-<3> 数组填充，比如A数组最后为【1，2，3】；那就填充【0，1，1，2，2，2】（1个0两个1三个2）

3、java代码

/**
     * 解决方案：计数排序
      * @param nums
     */
    public void sortColors(int[] nums) {
    //数组判空
      if (nums.length<=1){return ; }

    //1、定义一个常数空间的数组
      int[][] res=new int[][]{{0,0},{1,0},{2,0}};

    //2、遍历原数组，标记每个数组的个数
      for(int i=0;i<nums.length;i++){
          if(nums[i]==0){res[0][1]=res[0][1]+1;}
          if(nums[i]==1){res[1][1]=res[1][1]+1;}
          if(nums[i]==2){res[2][1]=res[2][1]+1;}
      }

    //3、给原数组赋值排序后的数组
        for(int j=0;j<nums.length;j++){
            if(j<res[0][1]){nums[j]=0;}
            else if(j>=res[0][1]&&j<res[1][1]+res[0][1]){nums[j]=1;}
            else {nums[j]=2;}
        }
    }