100道面试必会算法-35-合并区间

100道面试必会算法-35-合并区间

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

问题描述

给定一个区间的集合 intervals ，其中每个区间 [start, end] 都是闭区间。请合并所有重叠的区间，并返回合并后的区间集合。

算法思路

1. 首先对给定的区间集合 intervals 按照区间的起始位置进行排序，以确保相邻的区间能够被正确比较和合并。
2. 初始化一个新的数组 pre 来存储合并后的区间，长度为原始区间集合的长度。
3. 遍历排序后的区间集合 intervals，通过维护变量 end 表示当前合并后的区间的末尾位置，以及 count 表示遍历到的位置。
4. 将第一个区间放入 pre 数组。
5. 遍历 intervals中的每一个区间：
   • 如果当前区间与前一个区间有重叠，则将其合并到前一个区间。
   • 如果当前区间与前一个区间无重叠，则将其作为一个新的区间追加到 pre 数组中。
6. 最终返回合并后的数组，即 pre 数组的有效部分。

算法分析

• 时间复杂度：O(nlogn)，其中 n 是区间的个数，主要消耗在排序上。
• 空间复杂度：O(n)，需要额外的空间存储合并后的区间。

算法实现

class Solution {
    // 合并重叠区间
    public int[][] merge(int[][] intervals) {
        // 按区间起始位置排序
        Arrays.sort(intervals, (a, b) -> { return a[0] - b[0]; });
        
        // 初始化新数组
        int[][] pre = new int[intervals.length][intervals[0].length];
        
        int end = 0;
        int count = 1;
        // 将第一个区间放入pre数组
        pre[0] = intervals[0];
        
        while (count < intervals.length) {
            // 如果当前区间与前一个区间有重叠，则合并
            if (pre[end][1] >= intervals[count][0] && pre[end][1] < intervals[count][1]) {
                pre[end][1] = intervals[count][1];
            } else if (pre[end][1] >= intervals[count][0] && pre[end][1] >= intervals[count][1]) {
                count++;
                continue;
            } else {
                // 否则追加新的区间
                end++;
                pre[end] = intervals[count];
            }
            count++;
        }
        
        // 返回合并后的数组
        return Arrays.copyOfRange(pre, 0, end + 1);
    }
}