颜色分类(双“指针”)

题意:给定一个只包含0,1,2的数组,要求对这数组进行排序

思路:一个简单的思路:记录数组中0,1,2的个数,之后对数组进行重新赋值。该思路需要两次遍历。下面介绍一下只进行一次遍历的方法:

采用双“指针”,设立变量index0和index2记录0和2的位置。即区间[0,index0)都为0,区间(index2, len-1]都为2,其中len为数组长度。index0的初值为0,index2的初值为len-1

先做如下设定:输入数组nums,当前遍历位置nums[i]

解题策略如下:

1.nums[i] == 0时,nums[i]与nums[index0]互换,之后index0++,i++

2.nums[i] == 2 && index2 > i时, nums[i]与nums[index2]互换,之后index2--

3.nums[i] == 1时,i++

对于2需要做2点解释:

1).为什么限制index2 > i?

考虑情况:当数组已经排好序,如:[0,0,1,1,2,2]时,若不进行该限制,则会将已排好序的数组打乱。

2).为什么不进行i++?

因为index2所处的位置是没被遍历过的,若是进行了i++,则相当于跳过了一个元素,导致答案错误。

代码:

class Solution {
public:
    void sortColors(vector<int>& nums) {
        
        int len = nums.size();
        if(len <= 1) return;
        
        int index0 = 0, index2 = len - 1;
        for(int i=0;i<len;)
        {
            if(nums[i] == 2 && index2 > i)
            {
                nums[i] ^= nums[index2];
                nums[index2] ^= nums[i];
                nums[i] ^= nums[index2];
                index2--;
            }
            else if(nums[i] == 0)
            {
                nums[i] ^= nums[index0];
                nums[index0] ^= nums[i];
                nums[i] ^= nums[index0];
                index0++;
                i++;
            }
            else i++;
        }
        
        
    }
};

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值