Leetcode 75 颜色分类 C++

最新推荐文章于 2022-06-29 10:36:30 发布

原创最新推荐文章于 2022-06-29 10:36:30 发布 · 351 阅读

1 ·

CC 4.0 BY-SA版权

本文探讨了荷兰国旗问题的两种解决方案。第一种方法通过两次遍历数组，使用一维数组统计0、1、2的出现次数，然后在原数组上进行赋值。第二种方法仅遍历一次数组，利用两个指针分别指向数组头部和尾部，遇到0向前搬移，遇到2向后搬移，实现数组的0、1、2排序。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

解法一：思路：两次遍历数组。第一次遍历用一个一维数组统计0、1、2出现的个数。
第二次遍历数组在原有数组上进行赋值。

class Solution {
public:
    void sortColors(vector<int>& nums) {
        int count[3]={0},t=0;
        for(int i=0;i<nums.size();++i) count[nums[i]]++;
        for(int i=0;i<=2;++i)
            for(int j=0;j<count[i];++j)
            {
                nums[t++]=i;
            }
        
    }
};

解法二：
思路：遍历一次数组。用两个指针分别指向首尾。遍历一次数组，遇到0时，将0向前搬移，遇到2时，向后搬移。因为一共只有三个数，所以可以这样做。遍历一次数组后，数组就可以按0、1、2的顺序排列。

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