Leetcode 75.颜色分类——sort

最新推荐文章于 2023-10-16 23:05:44 发布

Arthur-Ji

最新推荐文章于 2023-10-16 23:05:44 发布

阅读量266

点赞数

分类专栏： LeetCode与数据结构

本文链接：https://blog.csdn.net/Arthur_Holmes/article/details/97621442

版权

LeetCode与数据结构专栏收录该内容

42 篇文章 1 订阅

订阅专栏

- 首先遍历一遍原数组，分别记录 0，1，2 的个数。
- 然后更新原数组，按个数分别赋上 0，1，2。

class Solution {
public:
    void sortColors(vector<int>& nums) {
        vector<int> colors(3);
        for (int num : nums) ++colors[num];
        for (int i = 0, cur = 0; i < 3; ++i) {
            for (int j = 0; j < colors[i]; ++j) {
                nums[cur++] = i;
            }
        }
    }
};