在做题中学习(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]);
            }
        }   
    }
};
相关推荐
吃好睡好便好5 小时前
提取矩阵某一行或某一列元素
开发语言·人工智能·线性代数·算法·matlab·矩阵
圣保罗的大教堂7 小时前
leetcode 2540. 最小公共值 简单
leetcode
wljy18 小时前
二、进制状态转换
linux·运维·服务器·c语言·c++
云泽8088 小时前
笔试算法 -位运算篇(二):从唯一字符到消失数字
c++·算法·位运算
ʚ希希ɞ ྀ8 小时前
不同路径|| -- dp
算法
繁华落尽,倾城殇?9 小时前
[C++11] : atomic,nullptr,default/delete,enum class
开发语言·c++·c++11·nullptr·atomic·enum class·default/delete
小新同学^O^9 小时前
简单学习 --> 模型参数
学习·llm·大模型参数
cdbqss19 小时前
VB2026 菜单生成基类 BqGetMenuStrip
数据库·经验分享·学习·oracle·vb
代码村新手9 小时前
C++-二叉搜索树
开发语言·c++
IT 行者9 小时前
SimHash 与 MinHash:相似性计算的双子星算法
算法·hash·比对