双指针 (leetcode 例题)

例一 颜色分类 (leetccode 75)

https://leetcode-cn.com/problems/sort-colors/icon-default.png?t=M0H8https://leetcode-cn.com/problems/sort-colors/

给定一个包含红色、白色和蓝色、共 n 个元素的数组 nums ,原地对它们进行排序,使得相同颜色的元素相邻,并按照红色、白色、蓝色顺序排列。

我们使用整数 0、 1 和 2 分别表示红色、白色和蓝色。

必须在不使用sort函数的情况下解决这个问题。

示例1:

示例2:

题目分析:

本题是经典的「荷兰国旗问题」,由计算机科学家 Edsger W. Dijkstra 首先提出。

根据题目中的提示,我们可以统计出数组中 0, 1, 2 的个数,再根据它们的数量,重写整个数组。这种方法较为简单,也很容易想到,而本题解中会介绍基于指针进行交换的方法。

因为只有 012 三种数字,所以我们划分出三种区域来存放这三种数字,存放完后即完成了顺序排序。

我们定义两个指针 q,p 分别作为区域的分界线,其中初始 q = 0, p = nums.size() - 1 。

从左往右遍历一遍,如果 nums[ i ] == 0 就与 q 所指向的元素交换位置,同时 q 指针往右移动一个单位;如果 nums[ i ] == 2 就与 p 所指向的元素交换位置,同时 p 往左移动一个单位,实现区域存放。

值得注意的是在我们将 nums[ i ] 交换位置后,此时新的 nums[ i ] 可能仍为 0 或 2 ,但指针已移动,所以我们不会再考虑新的 nums[ i ] ,以至于结果不是我们所想要的。

所以我们需先遍历一遍,找到所有的 2 ,这样新的 nums[ i ] 就不会是 2 ,这样我们只需要找 0 即可,0进行换位,如果是 1,那么就不需要进行任何后续的操作。

由于其中一个指针 p 是从右向左移动的,因此当我们在从左向右遍历整个数组时,如果遍历到的位置超过了 p,那么就可以直接停止遍历了。

看一下代码

题解代码:

class Solution {
public:
    void sortColors(vector<int>& nums) {
        int n = nums.size();

        int q = 0, p = n - 1;

        for(int i = 0; i <= p; i ++){
            while(i <= p && nums[i] == 2){
                swap(nums[i], nums[p]);

                p --;
            }// 先遍历一遍,找到所有的 2 

            if(nums[i] == 0){
                swap(nums[i], nums[q]);

                q ++;
            }
        }
    }
};

例二  救生艇  (leetcode 881)

https://leetcode-cn.com/problems/boats-to-save-people/icon-default.png?t=M0H8https://leetcode-cn.com/problems/boats-to-save-people/

给定数组 people 。people[ i ] 表示第 i 个人的体重 ,船的数量不限,每艘船可以承载的最大重量为 limit 。

每艘船最多可同时载两人,但条件是这些人的重量之和最多为 limit 。

返回 承载所有人所需的最小船数 。

示例1:

示例2:

示例3:

题目分析:

本题也运用了贪心思想

因为每艘船最多能载两个人,要使船数最少,应使载两人的船尽可能的多。

我们先对数组进行 sort 排序,这样我们能知道体重最轻和体重最重的人(数组头与数组尾),同时我们定义两个指针 q = 0 , p = people.size() - 1 分别指向最轻与最重,将这两者体重相加,若小于等于 limit ,我们就可将两者放到一个船上,这时将 q 向右移动,p 向左移动(得到新的最重与最轻);若大于 limit ,那么只能将最重的单独放到一条船上,这时只将 p 左移。

统计每次移动情况,即可得到最小船数。

题解代码:

class Solution {
public:
    int numRescueBoats(vector<int>& people, int limit) {
        sort(people.begin(), people.end());

        int n = people.size();
        int q = 0, p = n - 1;
        int count = 0;

        while(q <= p){
            if(people[q] + people[p] <= limit){
               q ++;
               p --;

               count ++;
            }else{
               p --;

               count ++;
            }
        }

        return count;
    }
};
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值