Leetcode 56. 合并区间

题目

给出一个区间的集合，请合并所有重叠的区间。

示例 1:

输入: [[1,3],[2,6],[8,10],[15,18]]
输出: [[1,6],[8,10],[15,18]]
解释: 区间 [1,3] 和 [2,6] 重叠, 将它们合并为 [1,6].

示例 2:

输入: [[1,4],[4,5]]
输出: [[1,5]]
解释: 区间 [1,4] 和 [4,5] 可被视为重叠区间。

解法

三个步骤：

1. 将所有的区间按照起始位置升序排序。
2. 判断当前区间的结束位置与后一个区间的起始位置的大小，若后者小于前者，说明可以合并。
3. 不断循环，直到结束。

代码

class Solution {
    public int[][] merge(int[][] intervals) {
    	// lambda 表达式虽然简洁，但是效率堪忧
        // Arrays.sort(intervals, (a1, a2) -> a1[0] - a2[0]);
        
        Arrays.sort(intervals, new Comparator<int[]>() {
            @Override
            public int compare(int[] a1, int[] a2) {
                return a1[0] - a2[0];
            }
        });
        
        List<int[]> res = new ArrayList<>();
        for(int i = 0; i < intervals.length; i ++) {
            int start = intervals[i][0];
            int end = intervals[i][1];
            // 不断合并区间
            while(i < intervals.length - 1 && intervals[i + 1][0] <= end) {
                end = Math.max(end, intervals[i + 1][1]);
                i ++;
            }
            
            res.add(new int[]{start, end});
        }
        
        // 声明长度为 0， 让内部返回一个新的数组
        return res.toArray(new int[0][]);                    
    }
}

结果