leetcode
56. 合并区间

问题描述

LeetCode 56. 合并区间 (opens in a new tab),难度中等

以数组 intervals 表示若干个区间的集合,其中单个区间为 intervals[i] = [starti, endi] 。请你合并所有重叠的区间,并返回 一个不重叠的区间数组,该数组需恰好覆盖输入中的所有区间

示例 1

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

示例 2

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

提示:

  • 1 <= intervals.length <= 104
  • intervals[i].length == 2
  • 0 <= starti <= endi <= 104

题解

Solution.java
class Solution {
    public int[][] merge(int[][] intervals) {
        // 数组根据第一个元素进行排序
        Arrays.sort(intervals, new Comparator<int[]>() {
            @Override
            public int compare(int[] a, int[] b) {
                // 根据第一个元素进行比较
                return Integer.compare(a[0], b[0]);
            }
        });
 
        int[][] result = new int[10001][];
        int index = -1;
        for (int i = 1; i < intervals.length; ++i) {
            if (intervals[i - 1][1] >= intervals[i][0]) {
                // 更新当前二维数组
                intervals[i][0] = Math.min(intervals[i - 1][0], intervals[i][0]);
                intervals[i][1] = Math.max(intervals[i - 1][1], intervals[i][1]);
            } else {
                // 加入到 result
                result[++index] = new int[2];
                result[index][0] = intervals[i - 1][0];
                result[index][1] = intervals[i - 1][1];
            }
        }
        // 最后一组数据更新了,但是为更新到数组
        result[++index] = new int[2];
        result[index][0] = intervals[intervals.length - 1][0];
        result[index][1] = intervals[intervals.length - 1][1];
        return Arrays.copyOfRange(result, 0, index + 1);
    }
}