leetcode 75

最新推荐文章于 2023-07-07 14:31:01 发布

love_4me

最新推荐文章于 2023-07-07 14:31:01 发布

阅读量191

点赞数

文章标签： leetcode 算法数据结构

本文链接：https://blog.csdn.net/weixin_45591968/article/details/129977214

版权

总体思路：这题还是利用桶排序，先统计出每个数字出现的数字频率，之后从小到大按照频率打印就行。

class Solution {
public:
    void sortColors(vector<int>& nums) {
        unordered_map<int, int> counts;
        for (const int & num : nums) {
            ++counts[num];
        }
        int num0 = counts[0];
        int num1 = counts[1];
        int num2 = counts[2];
        for (int i = 0; i < num0; i++) {
            nums[i] = 0;
        }
        for (int i = num0; i < num0 + num1; i++) {
            nums[i] = 1;
        }
        for (int i = num1 + num0; i < num0 + num1 + num2; i++) {
            nums[i] = 2;
        }
    }
};

注意：这样不符合题目空间复杂度的要求。

改：这其实是一个荷兰国旗问题，使用三指针来解决。

class Solution {
public:
    void sortColors(vector<int>& nums) {
        int l = 0, r = nums.size() - 1, index = 0;
        while (index <= r) {
            if (nums[index] == 0) {
                swap(nums[index], nums[l]);
                ++index;
                ++l;
            }
            else if (nums[index] == 1) {
                ++index;
            }
            else {
                swap(nums[index], nums[r]);
                --r;
            }
        }
    }
};

l是下一个要填0的位置，r是下一个要填2的位置。index是当前指针。则有[0,l-1]都是0，[r+1,n-1]都是2，[l,index - 1]都是1，[0,index]是已经处理过了的，然后[index+1,r]是还没有处理的。所以分成三种情况处理，如果index指向0，就将他与l指向的互换(l++,index++)，如果index指向1，则不用换，直接index后移即可，如果index指向2，则将他与r指向的互换，但只有r--,因为[index+1,r]未处理，所以从r换过来的不知道是什么，index不变还要再进循环进行判断。