荷兰国旗问题——LeetCode75

荷兰国旗问题——LeetCode75

荷兰国旗问题描述如下:

给定一个整数数组和一个值M(存在于原数组中),要求把数组中小于M的元素放到数组的左边,等于M的元素放到数组的中间,大于M的元素放到数组的右边,最终返回一个整数数组,只有两个值,0位置是等于M的数组部分的左下标值、1位置是等于M的数组部分的右下标值。

LeetCode75描述如下:

给定一个包含红色、白色和蓝色、共 n 个元素的数组 nums ,原地对它们进行排序,使得相同颜色的元素相邻,并按照红色、白色、蓝色顺序排列。
我们使用整数 0、 1 和 2 分别表示红色、白色和蓝色。
必须在不使用库内置的 sort 函数的情况下解决这个问题。

方法一:
使用辅助空间,分别记录数组中0、1以及2元素的个数

void sortColors(vector<int>& nums) {
        int colors[3];		//存储各个元素的数量
        for(int i = 0;i<3;i++){
            colors[i] = 0;
        }
        for(int i = 0;i<nums.size();i++){
            if(nums[i] == 0){
                colors[0]++;
            }
            else if(nums[i] == 1){
                colors[1]++;
            }
            else{
                colors[2]++;
            }
        }
        int xia = 0;		//指示颜色中的下标
        for(int k = 0;k<nums.size();k++){
            while(colors[xia]==0){
                xia++;
            }
            nums[k] = xia;
            colors[xia]--;
        }
    }

方法二:
遍历第一次,将0元素全部放在数组开头(使用交换);遍历第二次,将元素1接着放置(或者将元素2放置在数组末尾,一样的道理);那么剩下的2全部都在末尾(全部的1在中间)
方法三:
使用双指针优化方法二。p0指向下一个0应该存放的位置,p1指向下一个1应该存放的位置,还有一个指针用来遍历。遍历规则如下:
若遍历到1,将其与p1处元素交换;
若遍历到0,将其与p0处元素交换。但是此时可能导致原本的1被换到2“堆”中。所以还需要一步,将其再与p1处元素交换。

(待完成,补充图)

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值