【LeetCode热题100】【技巧】颜色分类

最新推荐文章于 2024-10-16 10:13:45 发布

MaolinYe（叶茂林）

最新推荐文章于 2024-10-16 10:13:45 发布

阅读量183

点赞数 4

分类专栏： # LeetCode热题 100 文章标签： leetcode 算法

叶茂林

本文链接：https://blog.csdn.net/weixin_62264287/article/details/137474923

版权

LeetCode热题 100 专栏收录该内容

100 篇文章 3 订阅

订阅专栏

题目链接：75. 颜色分类 - 力扣（LeetCode）

只需排序三种，可以记录0和1的个数，然后直接原地赋值

class Solution {
public:
    void sortColors(vector<int> &nums) {
        int zero = 0, one = 0;
        for (auto &num: nums)
            if (num == 0)
                zero++;
            else if (num == 1)
                one++;
        for (int i = 0; i < zero; i++)
            nums[i] = 0;
        for (int i = zero; i < zero + one; i++)
            nums[i] = 1;
        for (int i = zero + one; i < nums.size(); i++)
            nums[i] = 2;
    }
};