LeetCode 75 颜色分类(荷兰国旗问题,三数排序)

题目描述
给定一个包含红色、白色和蓝色,一共 n 个元素的数组,原地对它们进行排序,使得相同颜色的元素相邻,并按照红色、白色、蓝色顺序排列。此题中,我们使用整数 0、 1 和 2 分别表示红色、白色和蓝色。

示例 1:
输入:nums = [2,0,2,1,1,0]
输出:[0,0,1,1,2,2]
示例 2:
输入:nums = [2,0,1]
输出:[0,1,2]
示例 3:
输入:nums = [0]
输出:[0]
示例 4:
输入:nums = [1]
输出:[1]

题解
数轴上,起始位置 i=0,j=0,k=n-1,三数排序要保证的是[0,j-1] 区间内的数是0,[j,i-1] 区间是1,[k,n-1] 区间是2。排序的过程如下:

  • 如果nums[i]==0,就需要就将 i 指针所指向的数与 j 指针所指向的数交换,且由于交换过去了,j 已经是满足等于0的位置了,因此j++,i++
  • 如果nums[i]==1,那么i++即可,要满足[j,i-1] 区间,因此i++
  • 如果nums[i]==2,需要将2位置交换到k后面,因此k–,i 没有变化是因为交换过去的数字不确定是1还是0。
class Solution {
public:
    void sortColors(vector<int>& nums) {
        for(int i=0,j=0,k=nums.size()-1;i<=k;)
        {
            if(nums[i]==0) swap(nums[i++],nums[j++]);
            else if(nums[i]==2) swap(nums[i],nums[k--]);
            else if(nums[i]==1) i++;
        }
    }
};
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值