leecode-70-颜色分类-算法题解

0 引言

这道题要求在原地数组上进行操作,也就意味着不能新开一个数组,并且仅用常数的情况下去实现效果。因此我们需要在遍历数组的同时就要操作数组,从而实现排序的效果,此题的难点在于如何在遍历的同时进行排序。

1 方法

这段代码通过一次遍历实现对数组中 0、1、2 的排序,核心思想是用两个指针 jk 分别记录当前 1 和 2 的起始位置(或边界),在遍历过程中根据当前元素与前一个元素是否变化,动态调整元素位置:遇到 0、1、2 时分别插入到对应区域,并通过覆盖和前移的方式维持数组中 0 在前、1 在中、2 在后的顺序,从而实现类似"三色排序"的原地排序效果。

cpp 复制代码
class Solution{
    public:
        void sortColors(vector<int>& nums) {
            int i=0,j=-1,k=-1;
            if (nums.size()>1){
                for(int i=0;i<nums.size();i++){
                    if (i==0){
                        if (nums[i]==1){j=0;continue;}
                        else if(nums[i]==2){k=0;continue;}
                        else {continue;}
                    }
                    if (nums[i] != nums[i-1]){
                        if (nums[i] == 1){
                            if (j==-1 && k!=-1){
                                nums[i] = nums[i-1];
                                nums[k] = 1;
                                j = k;
                                k++;
                            }
                            else if(j==-1 && k==-1) j = i;
                            else{
                                nums[i] = nums[i-1];
                                nums[k] = 1;
                                k++;
                            }
                        }
                        else if (nums[i]==2){
                            if (k==-1) k = i;
                        }
                        else{
                            if (j==-1 && k==-1)continue;
                            else if(j==-1&&k!=-1){
                                nums[i] = nums[i-1];
                                nums[k] = 0;
                                k++;
                            }
                            else if(j!=-1 && k!=-1){
                                nums[i] = nums[i-1];
                                nums[j] = 0;
                                nums[k] = 1;
                                j++,k++;
                            }
                            else if (j!=-1 && k==-1){
                                nums[i] = nums[i-1];
                                nums[j] = 0;
                                j++;
                            }

                        }
                    }
                }
            }
    }
};
相关推荐
luj_17686 小时前
星火科技助力边远地区防病攻坚
c语言·开发语言·c++·经验分享·算法
木木子228 小时前
# 鸿蒙 ArkTS 实战:秒表 Stopwatch(示例 5)
数据结构·华为·list·harmonyos
geats人山人海9 小时前
数据结构第六章c语言 树的存储下二叉树的概念和存储
c语言·数据结构·算法
漂流瓶jz10 小时前
UVA-12627 奇怪的气球膨胀 题解答案代码 算法竞赛入门经典第二版
算法·图论·递归·aoapc·算法竞赛入门经典·uva·12627
jarvisuni10 小时前
DeepSeekFlash前端依旧拉垮,而且变慢了很多!
前端·javascript·算法
kobesdu10 小时前
流形上的优化:SO(3)与SE(3)的广义加减法在FAST-LIO中如何简化状态估计
人工智能·算法·fastlio
白狐_79810 小时前
408 数据结构算法题 01:线性表暴力求解保分指南
java·数据结构·算法
乐观勇敢坚强的老彭11 小时前
C++信奥:开关门、开关灯问题
开发语言·c++·算法
冻柠檬飞冰走茶11 小时前
PTA基础编程题目集 7-31 字符串循环左移(C语言实现)
c语言·开发语言·数据结构·算法
Hi李耶11 小时前
【LeetCode】4-寻找两个正序数组的中位数
算法·leetcode·职场和发展