LeetCode 75. 分类颜色

最新推荐文章于 2024-09-24 08:13:09 发布

一个假程序媛

最新推荐文章于 2024-09-24 08:13:09 发布

阅读量892

点赞数

分类专栏： LeetCode 文章标签： LeetCode 排序

本文链接：https://blog.csdn.net/sinat_15723179/article/details/80877667

版权

LeetCode 专栏收录该内容

54 篇文章 0 订阅

订阅专栏

思路：定义3个下标，一个start和一个end,分别指向vector的前面和后面，一个可以活动的下标cursor，主要思想就是把遍历到的0和start指向的值交换，把遍历得到的2和end指向的值交换，这样就确保了vector的前面都是0，后面都是2，那么自然就排好序了。

class Solution {
public:
    void swap(int &a, int &b){
        int temp=a;
        a=b;
        b=temp;
    }
    
    void sortColors(vector<int>& nums) {
        if(nums.size()==0 or nums.size()==1)
            return;
        
        int start=0, end=int(nums.size())-1, cursor=0;
        while(cursor<=end){
            if(nums[cursor]==0 && cursor==start){
                swap(nums[cursor], nums[start]);
                start++;
                cursor++; 
            }
            else if(nums[cursor]==0 && cursor!=start){
                swap(nums[cursor], nums[start]);
                start++;
            }
            else if(nums[cursor]==2){
                swap(nums[cursor], nums[end]);
                end--;
            }
            else
                cursor++;               
        }
    }
};