Leetcode—75.颜色分类【中等】

最新推荐文章于 2024-08-04 16:45:57 发布

源代码•宸

最新推荐文章于 2024-08-04 16:45:57 发布

阅读量480

点赞数 8

分类专栏： LeetCode刷题文章标签： leetcode 算法职场和发展经验分享 c++

本文链接：https://blog.csdn.net/qq_44631615/article/details/135119255

版权

LeetCode刷题专栏收录该内容

364 篇文章 5 订阅

订阅专栏

2023每日刷题（六十五）

Leetcode—75.颜色分类

在这里插入图片描述

实现代码

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