用最少数量的箭引爆气球的思路探讨与源码
用最少数量的箭引爆气球的题目如下图,该题属于贪心类型和排序类型的题目,主要考察对于排序方法的使用和贪心方法的理解。本文的题目作者想到2种方法,分别是贪心方法和重叠区间方法,其中贪心方法使用Java进行编写,而重叠区间方法使用Python进行编写,当然这可能不是最优的解法,还希望各位大佬给出更快的算法。
本人认为该题目可以使用贪心方法的思路进行解决,首先判断数组是否长度为0,如果是就直接返回0,然后定义一个排序的比对器,再取数组的第0行第1列的值并进行初始化赋值,然后开始遍历循环并找到大于初始值的元素并记录,直到遍历结束并返回最终结果。那么按照这个思路我们的Java代码如下:
#喷火龙与水箭龟
class Solution {
public int findMinArrowShots(int[][] points) {
if (points.length == 0) {
return 0;
}
Arrays.sort(points, new Comparator<int[]>() {
public int compare(int[] point1, int[] point2) {
if (point1[1] > point2[1]) {
return 1;
} else if (point1[1] < point2[1]) {
return -1;
} else {
return 0;
}
}
});
int ind = points[0][1];
int res = 1;
for (int[] balloon: points) {
if (balloon[0] > ind) {
ind = balloon[1];
res = res + 1;
}
}
return res;
}
}
显然,我们的贪心方法的效果不错,还可以用重叠区间方法进行解决。首先进行排序,再初始化列表,开始遍历循环,如果列表是空的就把元素加入到列表。如果列表是非空的,就把列表的元素和遍历值进行比较,如果列表末尾元素更小,则将列表进行弹出并比较记录最小值后插入到列表,以此逻辑进行遍历,直到遍历结束并返回列表的长度。所以按照这个思路就可以解决,下面是Python代码:
#喷火龙与水箭龟
class Solution:
def findMinArrowShots(self, points: List[List[int]]) -> int:
points.sort(key=lambda x: x[0])
stackList = []
for start, end in points:
if not stackList:
stackList.append([start, end])
else:
if start <= stackList[-1][-1]:
pre_start, pre_end = stackList.pop()
new_end = min(end, pre_end)
stackList.append([start, new_end])
else:
stackList.append([start, end])
return len(stackList)
从结果来说Java版本的贪心方法的效率不错,而Python版本的重叠区间方法的速度比较一般,但应该是有更多的方法可以进一步提速的,希望朋友们能够多多指教,非常感谢。