力扣hot100:75. 颜色分类(双指针)

75.颜色分类

本题是经典的「荷兰国旗问题」,由计算机科学家 Edsger W. Dijkstra 首先提出。

75. 颜色分类

1、遍历两遍

遍历两遍,第一遍放置0的位置,第二遍放置1的位置,我们只需要维护一个当前放置位置即可。

cpp 复制代码
class Solution {
public:
    void sortColors(vector<int>& nums) {//扫描两遍,第一遍放好0的位置,第二遍放1
        int zero = 0;
        for(int i = 0; i < nums.size(); ++ i){
            if(nums[i] == 0){
                swap(nums[i], nums[zero ++]);
            }
        }
        //复用zero
        for(int j = zero; j < nums.size(); ++ j){
            if(nums[j] == 1){
                swap(nums[j], nums[zero ++]);
            }
        }
        return;
    }
};

2、遍历一遍双指针

difficult to achieve

我们维护一个0的位置,并且维护一个1的位置:

  • 确保1的位置不小于0的位置
  • 当需要放入1时,直接加在当前指向需要放置的1的位置即可。
  • 当需要放入0时,如果后面有1,这会打断1,不过,我们可以把这个1再插入1的位置即可。如果没有1直接放入

这种方法比较难写,情况有点多,实现起来比较复杂。由于时间复杂度差不多,因此推荐使用简单方法实现!

cpp 复制代码
class Solution {
public:
    void sortColors(vector<int>& nums) {//扫描两遍,第一遍放好0的位置,第二遍放1
        int zero = 0;
        int one = 0;
        for(int i = 0; i < nums.size(); ++ i){//确保one在正确位置,情况分类比较难
            if(nums[i] == 1){
                swap(nums[i], nums[one ++]);
            }else if(nums[i] == 0){
                if(zero == one){
                    swap(nums[zero ++], nums[i]);
                    one ++;
                }else{
                    if(zero < one && zero != 0){//0有数,1也有数
                        swap(nums[one ++], nums[i]);
                        nums[one - 1] = 1;
                        nums[zero ++] = 0;
                    }else{//0没数
                        if(one != i){
                            nums[zero ++] = 0;
                            swap(nums[one], nums[i]);
                            nums[one] = 1;
                        }else{
                            swap(nums[zero ++],nums[i]);
                        }
                        one++;
                    }
                }
            }
        }
        return;
    }
};

3、双指针交换2的位置

cpp 复制代码
class Solution {
public:
    void sortColors(vector<int>& nums) {
        int n = nums.size();
        int p0 = 0, p2 = n - 1;
        for (int i = 0; i <= p2; ++i) {
            while (i <= p2 && nums[i] == 2) {
                swap(nums[i], nums[p2]);
                --p2;
            }
            if (nums[i] == 0) {
                swap(nums[i], nums[p0]);
                ++p0;
            }
        }
    }
};
相关推荐
SweetCode几秒前
裴蜀定理:整数解的奥秘
数据结构·python·线性代数·算法·机器学习
ゞ 正在缓冲99%…13 分钟前
leetcode76.最小覆盖子串
java·算法·leetcode·字符串·双指针·滑动窗口
xuanjiong14 分钟前
纯个人整理,蓝桥杯使用的算法模板day2(0-1背包问题),手打个人理解注释,超全面,且均已验证成功(附带详细手写“模拟流程图”,全网首个
算法·蓝桥杯·动态规划
惊鸿.Jh33 分钟前
【滑动窗口】3254. 长度为 K 的子数组的能量值 I
数据结构·算法·leetcode
明灯L34 分钟前
《函数基础与内存机制深度剖析:从 return 语句到各类经典编程题详解》
经验分享·python·算法·链表·经典例题
碳基学AI40 分钟前
哈尔滨工业大学DeepSeek公开课:探索大模型原理、技术与应用从GPT到DeepSeek|附视频与讲义免费下载方法
大数据·人工智能·python·gpt·算法·语言模型·集成学习
补三补四43 分钟前
机器学习-聚类分析算法
人工智能·深度学习·算法·机器学习
独好紫罗兰1 小时前
洛谷题单3-P5718 【深基4.例2】找最小值-python-流程图重构
开发语言·python·算法
正脉科工 CAE仿真1 小时前
基于ANSYS 概率设计和APDL编程的结构可靠性设计分析
人工智能·python·算法
Dovis(誓平步青云)2 小时前
【数据结构】排序算法(中篇)·处理大数据的精妙
c语言·数据结构·算法·排序算法·学习方法