【LeetCode】75. 颜色分类

最新推荐文章于 2025-08-17 20:47:32 发布

fuqiuai

最新推荐文章于 2025-08-17 20:47:32 发布

阅读量1.8k

点赞数

CC 4.0 BY-SA版权

分类专栏： LeetCode

本文链接：https://blog.youkuaiyun.com/fuqiuai/article/details/83349815

LeetCode 专栏收录该内容

118 篇文章

订阅专栏

本文详细解析了LeetCode上一道经典排序题目——三色排序，即对数组中的红、白、蓝三种颜色进行原地排序。文章介绍了计数排序算法的具体实现过程，包括通过一趟扫描计算各颜色数量，再按顺序重写数组，实现常数空间复杂度下的高效排序。

题目链接：https://leetcode-cn.com/problems/sort-colors/description/

题目描述

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

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

示例

输入: [2,0,2,1,1,0]
输出: [0,0,1,1,2,2]

进阶：

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

解决方法

计数排序

class Solution {
public:
    void sortColors(vector<int>& nums) {
        //计数排序
        vector<int> jishu(3,0);
        for (int i=0;i<nums.size();i++)
            jishu[nums[i]]++;
        cout<<jishu[0]<<" "<<jishu[1]<<" "<<jishu[2];
        for (int i=0;i<jishu[0];i++)
            nums[i]=0;
        for (int i=jishu[0];i<jishu[1]+jishu[0];i++)
            nums[i]=1;
        for (int i=jishu[1]+jishu[0];i<jishu[2]+jishu[1]+jishu[0];i++)
            nums[i]=2;
    }
};