从头做leetcode之leetcode 75 颜色分类

最新推荐文章于 2024-08-09 22:11:04 发布

原创最新推荐文章于 2024-08-09 22:11:04 发布 · 127 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode #算法 #快速排序

从头做leetcode 专栏收录该内容

94 篇文章

订阅专栏

本文深入解析了一种高效的颜色分类算法，特别适用于处理包含红、白、蓝三色元素的数组。文章介绍了两种实现方法，包括两遍扫描法和更优的一遍扫描法，后者能在一次遍历中完成排序，极大提升了效率。

75.颜色分类

- 两遍扫描
- 一遍扫描

给定一个包含红色、白色和蓝色，一共 n 个元素的数组，原地对它们进行排序，使得相同颜色的元素相邻，并按照红色、白色、蓝色顺序排列。

此题中，我们使用整数 0、 1 和 2 分别表示红色、白色和蓝色。

注意:
不能使用代码库中的排序函数来解决这道题。

两遍扫描

用三个变量记录三个颜色的数量，时间复杂度O(2n)

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

通过时间：
在这里插入图片描述

一遍扫描

扫描过程中将0放在左边，2放在右边。

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