LeetCode - 贪心 - 452. 用最少数量的箭引爆气球

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题库的贪心算法标签。

解题思路：
首先看题、分析题意：

1. 用最少的箭射穿最多的气球
2. 气球是有区间宽度的且会重合，尽可能找到重合的区域
   既然，我们已经分析出来题目的关键任务了，下面我们就可以开始思考实现了。
   我们采用算法与数据结构的思路来剖析一下这题

数据结构：
要实现对数据的操作，我们要先明确存储数据的数据结构。

• 二维数组
• 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]则表示为降序。

算法：

1. 将气球先按照区间的左端点进行升序排序
2. 维持射击区间：寻找尽可能多的重合区间，若下一个区间无法重合，则必须增加弓箭手

代码：

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;
    }
}