leetcode 75颜色分类

最新推荐文章于 2025-09-06 12:56:54 发布

转载最新推荐文章于 2025-09-06 12:56:54 发布 · 74 阅读

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/joelwang/p/10931303.html

博客介绍了排序算法，提到两趟扫描时因排序变量特殊性，用计数排序可将复杂度降至O(n) time O(n) space，还给出计数排序相关链接。此外，介绍使用3个变量一趟扫描的方法，复杂度为O(1) space O(n) time，并给出具体实现过程链接。

两趟扫描,由于排序变量的特殊性，使用计数排序方法可以明显降低至O(n)time O(n) space

关于计数排序：https://mp.weixin.qq.com/s/WGqndkwLlzyVOHOdGK7X4Q

class Solution {
public:
    void sortColors(vector<int>& nums) {
        if(nums.size()==0) return;
        int len=nums.size();
        vector<int> arr(3,0);
        for(int num:nums)
            arr[num]++;
        int index=0;
        for(int i=0;i<arr.size();i++){
            int k=arr[i];
            while(k--){
                nums[index++]=i;
            }
        }
    }
};

使用3个变量一趟扫描O(1) space O(n) time

/**
使用3个变量来分别表示3个颜色；
**/

class Solution {
public:
    void sortColors(vector<int>& nums) {
        if(nums.size()==0) return;
        int len=nums.size();
        int L=0,M=0,H=len-1;
        while(M<=H){
            if(nums[M]==1){
                M++;continue;
            }
            if(M<=H&&nums[M]==0){
                swap(nums[M],nums[L]);
                L++;M++;
            }
            if(M<=H&&nums[M]==2){
                swap(nums[M],nums[H]);
                H--;
            }
        }
    }
};

具体实现过程参见：https://leetcode.com/problems/sort-colors/discuss/26474/Sharing-C%2B%2B-solution-with-Good-Explanation

转载于:https://www.cnblogs.com/joelwang/p/10931303.html