56.合并区间(java)

题目描述

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

示例 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] 可被视为重叠区间。

解法一：排序

• 先将二维数组排序，遍历一下情况
  • 若当前答案区间最大值>当前遍历数组最大值，continue。
  • 若当前答案区间最大值>=当前遍历数组最小值，答案最大值更新为遍历数组最大值。
  • 若当前答案区间最大值<当前遍历数组最小值，创建新的答案组。

class Solution {
    public int[][] merge(int[][] intervals) {
        int n=intervals.length;
        if(n==0||n==1) return intervals;
        int[][] ans=new int[n][2];
        int index=0;
        Arrays.sort(intervals,(a,b)->a[0]-b[0]);
        ans[index][0]=intervals[0][0];
        ans[index][1]=intervals[0][1];
        for(int i=1;i<n;i++){
            if(ans[index][1]>=intervals[i][1]) continue;
            if(ans[index][1]>=intervals[i][0]){
                ans[index][1]=intervals[i][1];
            }
            else{
                index++;
                ans[index][0]=intervals[i][0];
                ans[index][1]=intervals[i][1];
            }
        }
        return Arrays.copyOf(ans,index+1);
    }
}

代码优化

class Solution {
    public int[][] merge(int[][] intervals) {
        int n=intervals.length;
        if(n==0||n==1) return intervals;
        int[][] ans=new int[n][2];
        Arrays.sort(intervals,(a,b)->a[0]-b[0]);
        int index=-1;
        for(int[] interval:intervals){
            if(index==-1||ans[index][1]<interval[0]){
                ans[++index]=interval;
            }
            else{
                ans[index][1]=Math.max(interval[1],ans[index][1]);
            }
        }
        return Arrays.copyOf(ans,index+1);
    }
}

总结技巧：

排序二维数组：(v1, v2) -> v1[0] - v2[0]

Arrays.sort(nums,(a,b)->a[0]-b[0]);

截断二维数组

Arrays.copyOf(ans,length);

列表转数组

List<int[]> ans=new ArrayList<>();
……
return ans.toArray(new int[0][]);