排序75.颜色分类

问号byr

于 2020-08-31 23:57:42 发布

阅读量168

点赞数

分类专栏： leetcode 文章标签： leetcode

本文链接：https://blog.youkuaiyun.com/weixin_39451679/article/details/108332043

版权

leetcode 专栏收录该内容

20 篇文章

订阅专栏

本文深入解析了一种针对红、白、蓝三种颜色的数组排序算法，通过计数排序和三指针法实现原地排序，确保相同颜色元素相邻并按顺序排列。文中提供了详细的代码示例，展示了如何在不使用标准库排序函数的情况下解决这一问题。

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

75.颜色分类

给定一个包含红色、白色和蓝色，一共 n 个元素的数组，原地对它们进行排序，使得相同颜色的元素相邻，并按照红色、白色、蓝色顺序排列。
此题中，我们使用整数 0、 1 和 2 分别表示红色、白色和蓝色。
注意:
不能使用代码库中的排序函数来解决这道题。
示例:
输入: [2,0,2,1,1,0]
输出: [0,0,1,1,2,2]

计数排序

class Solution {
public:
    void sortColors(vector<int>& nums) {
        int n = nums.size();
        int j=0,t=0,k=0;

        for(int i=0;i<n;i++)
        {
            if(nums[i]==0)j++;
            else if(nums[i]==1)t++;
            else if(nums[i]==2)k++;
        }
        for(int i=0;i<n;i++)
        {
            if(j>0)
            {
                nums[i]=0;
                j--;
                continue;
            }
            else if(t>0)
            {
                nums[i]=1;
                t--;
                continue;
            }
            else
            {
                nums[i]=2;
                k--;
                continue;
            }
        }
    }
};

三指针法一次遍历

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

  
    }
};