首页 > 技术文章 > 用最少数量的箭引爆气球-经典贪心

chengejie 2020-11-23 20:16 原文

一、问题描述

简述

在二维空间中有许多球形的气球。对于每个气球,提供的输入是水平方向上,气球直径的开始和结束坐标。由于它是水平的,所以纵坐标并不重要,因此只要知道开始和结束的横坐标就足够了。开始坐标总是小于结束坐标。

一支弓箭可以沿着 x 轴从不同点完全垂直地射出。在坐标 x 处射出一支箭,若有一个气球的直径的开始和结束坐标为 xstart,xend, 且满足  xstart ≤ x ≤ xend,则该气球会被引爆。可以射出的弓箭的数量没有限制。 弓箭一旦被射出之后,可以无限地前进。我们想找到使得所有气球全部被引爆,所需的弓箭的最小数量。

输入

6
10 16 2 8 1 6 7 12

输出

2

解释:对于该样例,x = 6 可以射爆 [2,8],[1,6] 两个气球,以及 x = 11 射爆另外两个气球

注意测试数据

1
-2147483648 2147483647

二、解题思路

  1. 首先按照气球尾部大小从小到大排序
  2. 循环开始从第一个气球的尾部射出一箭,即k = b[i].end-1;
  3. 然后从最后一个被射穿的气球后一个开始循环,即b[i].start>=k
  4. 这样能保证每次射穿的气球是最多的,即需要的弓箭是最少的。

三、代码

#include <iostream>
#include <algorithm>

using namespace std;

struct Balloon{
    int start,end;
};
Balloon *b;
bool cmp(Balloon t1,Balloon t2){
    return t1.end<t2.end;
}
int main()
{
    int n;
    cin>>n;
    b = new Balloon[n];
    for (int i = 0; i < n; ++i) {
        cin>> b[i].start>>b[i].end;
    }
    sort(b,b+n,cmp);
    int sum = 0;
    int k = -1<<31;
    for (int i = 0; i < n; ++i) {
        if(b[i].start>=k){
            k = b[i].end-1;
            sum++;
        }
    }
    cout<<sum<<endl;
    return 0;
}

推荐阅读