Leetcode刷题笔记(c++)_热题 HOT 100_75. 颜色分类

最新推荐文章于 2024-08-21 10:11:50 发布

大风吹~~~~~

最新推荐文章于 2024-08-21 10:11:50 发布

阅读量266

点赞数

文章标签：多指针计数排序快速排序彩色球问题数组操作

本文链接：https://blog.csdn.net/LW_12345/article/details/125379215

版权

多指针

class Solution {
public:
    void sortColors(vector<int>& nums) {
        int left=0,right=nums.size()-1,mid=left+1;
        while(left<right&&mid<=right){
            // cout<<left<<" "<<mid<<" "<<right<<endl;
            // cout<<nums[left]<<" "<<nums[mid]<<" "<<nums[right]<<endl;
            if(nums[left]==0||nums[right]==2){
                if(nums[left]==0){
                    left++;
                    if(mid<=left)mid=left+1;
                }
                if(nums[right]==2)right--;                
            }else if(nums[left]>nums[mid])swap(nums[left],nums[mid]);
            else if(nums[mid]>nums[right])swap(nums[right],nums[mid]);
            else mid++;            
        }             
    }
};

在这里插入图片描述

计数

class Solution {
public:
    void sortColors(vector<int>& nums) {
        vector<int>v(3,0);
        for(int i:nums)v[i]++;
        // cout<<v[0]<<" "<<v[1]<<" "<<v[2]<<" ";
        for(int i=0;i<v[0];i++)nums[i]=0;
        for(int i=v[0];i<v[0]+v[1];i++)nums[i]=1;
        for(int i=v[0]+v[1];i<v[0]+v[1]+v[2];i++)nums[i]=2;        
    }
};