力扣56 合并区间 Java版本

---

题目描述

以数组 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

代码

class Solution {
    public int[][] merge(int[][] intervals) {
        int[][] result = new int[intervals.length][2];
        //先把数组升序排列
        Arrays.sort(intervals,(a,b)->Integer.compare(a[0],b[0]));
        int start = intervals[0][0];
        int end = intervals[0][1];
        int num = 0;
        if (intervals.length==1){
            return intervals;
        }
        for (int i = 1; i < intervals.length; i++) {
            //如果发现又重叠的部分就更新一下end，让end把两段内容都包括进去
            if (intervals[i][0]<=end){
                //第二段区间有可能含于第一段区间也有可能是相交，所以取最大值
                end = Math.max(end,intervals[i][1]);
                if (i==intervals.length-1){
                    result[num][0] = start;
                    result[num][1] = end;
                    num++;
                }
            }else {
                //如果不重叠就把这一段区间保存到结果当中，并且更新新的start和end
                result[num][0] = start;
                result[num][1] = end;
                num++;
                start = intervals[i][0];
                end = intervals[i][1];
                if (i==intervals.length-1){
                    result[num][0] = start;
                    result[num][1] = end;
                    num++;
                }
            }
        }
        //返回结果中记录的区间
        int[][] res = Arrays.copyOfRange(result, 0, num );
        return res;
    }
}

优化后的代码

class Solution {
    public int[][] merge(int[][] intervals) {
        List<int[]> res = new LinkedList<>();
        Arrays.sort(intervals, (x, y) -> Integer.compare(x[0], y[0]));
        int start = intervals[0][0];
        int end = intervals[0][1];

        for (int i = 1; i < intervals.length; i++) {
            if (intervals[i][0]<=end){
                end = Math.max(intervals[i][1],end);
            }else {
                res.add(new int[]{start,end});
                start = intervals[i][0];
                end = intervals[i][1];
            }
        }
        res.add(new int[]{start,end});
        return res.toArray(new int[res.size()][]);
    }
}