在做题中学习(76):颜色分类

解法:三指针

思路:用三个指针,把数组划分为三个区域:

for循环遍历数组,i遍历数组,left是0区间的末尾,right是2区间的开头,0 1 2区间成功被划分

而上面的图画是最终实现的图样,实际left,right的移动得配合numsi来:

left = -1,right = nums.size();

1.numsi == 0,要保证0,left为全0,left++,swap(numsleft,numsi) ,处理完当前numsi,i++

合并为:swap(nums++left,numsi++

2.numsi == 1, 要保证left+1,i-1是全1,所以 i++ 即可保证。

3.numsi == 2,要保证right,n-1是全2,right--,swap(numsright,nums\[i),处理完当前numsi,不用i++,因为此时swap过去的numsrigjt 这个数依旧还没判断呢,所以此时不能i++

numsi == 0的例子i++是因为俩交换的数都已经判断完了。

合并为:swap(nums--right,numsi)

注意:结束标志,当i遍历时,i == right的时候,证明已经排好序了,结束循环即可。

附上思路图:

完整代码:

cpp 复制代码
class Solution {
public:
    void sortColors(vector<int>& nums) 
    {
        int n = nums.size(),l = -1,r = n;
        for(int i = 0;i<n;)
        {
            if(nums[i] == 0)
                swap(nums[++l],nums[i++]);
            else if(nums[i] == 1)
                i++;
            else if(nums[i] == 2)
            {
                if(i == r) break;
                swap(nums[--r],nums[i]);
            }
        }   
    }
};
相关推荐
Robot_Nav几秒前
贪心算法、动态规划与 MPPI 算法结构相关力扣题目汇总
算法·贪心算法·动态规划
B8017913Y10 分钟前
上课记笔记写不完不会整理?2026学习笔记生成使用场景该怎么选
人工智能·笔记·学习
战族狼魂34 分钟前
每日一课:算法系统学习路线
人工智能·算法·大模型·大语言模型
变量未定义~43 分钟前
连通块中点的数量、堆箱子(4星)
算法
盐焗鹌鹑蛋1 小时前
【C++】set和map
c++
j7~1 小时前
【算法】专题二:滑动窗口之水果成蓝,找到字符串中所有字⺟异位词等算法题
c++·算法·滑动窗口·水果成蓝·最小字串覆盖·优选算法精选
木木子221 小时前
# 星座配对深度解析:Select 长列表选择器、矩阵评分算法与数据驱动建议
java·算法·华为·矩阵·harmonyos
一拳一个呆瓜1 小时前
【STL】iostream 编程:文件输出流的成员函数
c++·stl
小园子的小菜1 小时前
Java设计模式精讲:创建型+结构型模式(含生产级案例+相似模式对比)
学习·设计模式
大鱼>1 小时前
AI+货物追踪:跨境电商包裹追踪系统
人工智能·深度学习·算法·机器学习