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

代码

/*
 * @lc app=leetcode.cn id=56 lang=java
 *
 * [56] 合并区间
 */

// @lc code=start
class Solution {
    public int[][] merge(int[][] intervals) {

        if(intervals == null || intervals.length < 2) {
            return intervals;
        }
        
        int n = intervals.length;
        Arrays.sort(intervals, (o1, o2) -> o1[0] - o2[0]);
        
        // 遍历所有区间，如果当前的起点在上一个末尾之前，则合并
        //              如果在上一个区间末尾之后，则把上一个区间放到答案中去
        List ans = new ArrayList<>();
        int[] pre = intervals[0];
        for(int i = 1; i < n; i++) {
            if(intervals[i][0] <= pre[1]) {
                pre[1] = Math.max(pre[1], intervals[i][1]);
            } else {
                ans.add(pre);
                pre = intervals[i];
            }
        }
        
        // 把最后一个放进去
        ans.add(pre);
        
        return ans.toArray(new int[ans.size()][2]);

    }
}
// @lc code=end