452. 用最少数量的箭引爆气球
在二维空间中有许多球形的气球。对于每个气球,提供的输入是水平方向上,气球直径的开始和结束坐标。由于它是水平的,所以y坐标并不重要,因此只要知道开始和结束的x坐标就足够了。开始坐标总是小于结束坐标。平面内最多存在104个气球。
一支弓箭可以沿着x轴从不同点完全垂直地射出。在坐标x处射出一支箭,若有一个气球的直径的开始和结束坐标为 xstart,xend, 且满足 xstart ≤ x ≤ xend,则该气球会被引爆。可以射出的弓箭的数量没有限制。 弓箭一旦被射出之后,可以无限地前进。我们想找到使得所有气球全部被引爆,所需的弓箭的最小数量。
Example:
输入: [[10,16], [2,8], [1,6], [7,12]]
输出: 2
解释:
对于该样例,我们可以在x = 6(射爆[2,8],[1,6]两个气球)和 x = 11(射爆另外两个气球)。
简单介绍:
难度:中等
使用语言:JAVA。
这道题来自leetcode题库的贪心算法标签。
解题思路:
首先看题、分析题意:
- 用最少的箭射穿最多的气球
- 气球是有区间宽度的且会重合,尽可能找到重合的区域
既然,我们已经分析出来题目的关键任务了,下面我们就可以开始思考实现了。
我们采用算法与数据结构的思路来剖析一下这题
数据结构:
要实现对数据的操作,我们要先明确存储数据的数据结构。
- 二维数组
- Array.sort()
补充知识:java中用sort进行自定义排序
int [][]a = new int [5][2];
定义一个二维数组,其中所包含的一维数组具有两个元素
对于一个已定义的二位数组a进行如下规则排序,首先按照每一个对应的一维数组第一个元素进行升序排序(即a[][0]),若第一个元素相等,则按照第二个元素进行升序排序(a[][1])。(特别注意,这里的a[][0]或者a[][1]在java中是不能这么定义的,这里只是想说明是对于某一个一维数组的第0或1个元素进行排序)
Arrays.sort(a, new Comparator<int[]>() {
@Override
public int compare(int[] o1, int[] o2) {
if (o1[0]==o2[0]) return o1[1]-o2[1];
return o1[0]-o2[0];
}
});
其中o1[1]-o2[1]表示对于第二个元素进行升序排序如果为o2[1]-o1[1]则表示为降序。
算法:
- 将气球先按照区间的左端点进行升序排序
- 维持射击区间:寻找尽可能多的重合区间,若下一个区间无法重合,则必须增加弓箭手
代码:
class Solution {
public int findMinArrowShots(int[][] points) {
if(points.length == 0)
return 0;
//自定义排序
Arrays.sort(points, new Comparator<int[]>() {
public int compare(int[] o1, int[] o2) {
if (o1[0]==o2[0])
return o1[1]-o2[1];
return o1[0]-o2[0];
}
});
//初始化弓箭手和射击区间
int shootnum = 1;
int shootstat = points[0][0];
int shootend = points[0][1];
for(int i = 1; i < points.length; i++){
//如果下一个气球的左端点在射击区间内,则更新射击区间的左端点
if(points[i][0] <= shootend){
shootstat = points[i][0];
//如果下一个气球的右端点在射击区间内,则更新射击区间的右端点
if(points[i][1] <= shootend){
shootend = points[i][1];
}
}else{
//如果下一个气球左端点不在射击区间内,说明已经超出了射击区间,则必须增加弓箭手
shootnum++;
shootstat = points[i][0];
shootend = points[i][1];
}
}
return shootnum;
}
}
优化代码
区间的末尾端点很重要:如果不使用新的箭,新区间末尾端点就要和当前的“最远距离”(含义通过上文可以明白)作一个比较,取最小值。那我不妨就按照区间的末尾端点排序,这样如果不使用新的箭,也不用作比较了
import java.util.Arrays;
import java.util.Comparator;
public class Solution {
public int findMinArrowShots(int[][] points) {
int len = points.length;
if (len < 2) {
return len;
}
// 按照区间终点进行排序
Arrays.sort(points, new Comparator<int[]>() {
@Override
public int compare(int[] point1, int[] point2) {
if (point1[1] != point2[1]) {
return point1[1] - point2[1];
}
return point1[0] - point2[0];
}
});
int count = 1;
int end = points[0][1];
for (int i = 1; i < len; i++) {
if (points[i][0] > end) {
// 就得多用一支箭
end = points[i][1];
count++;
}
}
return count;
}
}