LeetCode每日一题 (33)75. 颜色分类

最新推荐文章于 2024-09-09 09:35:59 发布

战胜.

最新推荐文章于 2024-09-09 09:35:59 发布

阅读量139

点赞数

分类专栏：战胜LeetCode刷题文章标签：排序算法颜色分类树中距离之和动态规划优化方法

本文链接：https://blog.csdn.net/qq_45021180/article/details/108953815

版权

战胜LeetCode刷题专栏收录该内容

63 篇文章 0 订阅

订阅专栏

75. 颜色分类

在这里插入图片描述

使用库函数(测试)

class Solution {
public:
    void sortColors(vector<int>& nums) {
        sort(nums.begin(),nums.end());
    }
};

在这里插入图片描述

统计个数，再赋值：

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

在这里插入图片描述

上面方法的改进：

class Solution {
public:
    void sortColors(vector<int>& nums) {
    //start表示0的开始位置，count1表示1的个数，count2表示2的个数
        int start=0,count1=0,count2=0; 
        for(int i=0;i<nums.size();i++){
            if(nums[i]==0){ //因为0在最前面，所有可以先将0的值写入
                nums[start]=0;
                start++;
            }
            else if(nums[i]==1) count1++;
            else count2++;
        }
        for(int i=start;i<nums.size();i++){
            if(i<count1+start) nums[i]=1;
            else nums[i]=2;
        }
    }
};