leetcode 75. 颜色分类

最新推荐文章于 2025-12-19 16:34:09 发布

原创最新推荐文章于 2025-12-19 16:34:09 发布 · 148 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode #算法 #c++ #数据结构

题目如下
在这里插入图片描述
数据范围

两趟扫描第一趟统计0 1 2的个数，
第二趟直接原地修改。

通过代码

class Solution {
public:
    void sortColors(vector<int>& nums) {
        int a = 0,b = 0,c = 0;
        int n = nums.size();
        for(int i = 0;i < n;i++){
            if(nums[i] == 0)a++;
            else if(nums[i] == 1)b++;
            else c++;
        }
        for(int i = 0;i < a;i++)nums[i] = 0;
        for(int i = a;i < a + b;i++)nums[i] = 1;
        for(int i = a + b;i < n;i++)nums[i] = 2;
    }
};