56 -合并区间 - python + 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] 可被视为重叠区间。

题目要求合并数组中所有重叠的区间，如果两两观察数组中的元素，如果两个元素有重叠，那么后一个元素的开始位置必然在第一个元素表示的范围内，即满足nextStart <= end。因此，首先将数组按字典序排列，使其元素的起始位置升序排列。然后进行判断合并，合并后的区间开始位置为第一个元素的起始位置，区间的结束位置为两个元素结束位置中的最大值，因为存在第二个元素完全位于第一个元素的区间内。

class Solution:
    def merge(self, intervals: List[List[int]]) -> List[List[int]]:
        if not intervals: return []
        # 先将数组按字典序排序
        intervals = sorted(intervals)

        # 两两比较
        start, end = intervals[0]
        res = []
        for i in range(1, len(intervals)):
            curStart, curEnd = intervals[i]
            if curStart <= end:
                end = max(end, curEnd)
            else:
                res.append([start, end])
                start = curStart
                end = curEnd

        res.append([start, end])

        return res

2020 - 7- 19 更新Java AC Code，如下所示：

/**
 * @Author dyliang
 * @Date 2020/7/19 22:32
 * @Version 1.0
 */
public class MergeArrays {
    public static void main(String[] args) {
//        int[][] intervals = {{1, 3}, {2, 6}, {8, 10}, {15, 18}};
        int[][] intervals = {{1, 4}, {5, 6}};
        int[][] res = merge(intervals);
        System.out.println("--------------------");
        for (int[] r : res) {
            System.out.println(Arrays.toString(r));
        }

    }

    public static int[][] merge(int[][] intervals) {
        if (intervals.length == 0 || intervals.length == 1){
            return intervals;
        }
        List<int[]> list = new ArrayList<>();

        // 首先按照起始位置升序排列
        Arrays.sort(intervals, new Comparator<int[]>() {
            @Override
            public int compare(int[] o1, int[] o2) {
                return o1[0] - o2[0];
            }
        });
        // 或使用函数式编程风格
        // Arrays.sort(intervals, (o1, o2) -> o1[0] - o2[0]);

        int start = intervals[0][0];
        int end = intervals[0][1];
        int index = 1;

        while (index < intervals.length){
            System.out.println(Arrays.toString(intervals[index]));
            int curStart = intervals[index][0];
            int curEnd = intervals[index][1];

            if (curStart <= end){
                end = Math.max(end, curEnd);
            } else {
                list.add(new int[]{start, end});
                start = curStart;
                end = curEnd;
            }
            index += 1;
        }
        
        list.add(new int[]{start, end});
        return list.toArray(new int[0][]);
    }
}