[LeetCode] 56、合并区间

题目描述

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

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

解题思路

先对区间按起点坐标排序. 这样, 能合并的区间都会聚集在一起。然后遍历每个区间[a,b]. 当a落在前面的区间中, 则[a,b]与前面的区间相合并，并更新新区间的结束坐标.

c++版思路：

intervals;区间列表
sort(intervals);
ans;结果区间列表
for(i in [0,...,N-1]){
    if(ans空 || ans.tail[1] < intervals[i][0]){
        intervals[i]入ans尾部;
    }else{
        ans.tail与intervals[i]合并;
    }
}

python版思路：

def merge(self, intervals: List[List[int]]) -> List[List[int]]:
    #先把intervals按照第一个值进行排序
    intervals.sort()
    n = len(intervals)
    if n == 0:
        return [] 
    result = [intervals[0]]
    for i in range(1,n):
        #此时必有重合，则重合区间的右值为最大的那个
        if intervals[i][0] <= result[-1][1]:
            result[-1][1] = max(result[-1][1],intervals[i][1])
        #无重合则直接添加区间
        else:
            result.append(intervals[i])
    return result

参考代码

class Solution {
public:
    vector<vector<int>> merge(vector<vector<int>>& intervals) {
        vector<vector<int>> res;
        if(intervals.size() == 0 || intervals[0].size() != 2)
            return res;
        
        sort(intervals.begin(), intervals.end(), cmp);
        
        for(int i = 0; i < intervals.size(); i++){
            if(res.empty() || intervals[i][0] > res.back()[1]){  // 此题 [[1,4],[4,5]] 也算重合（跟面试官问清楚）
                res.push_back(intervals[i]);
            }else{
                vector<int> temp;
                temp.push_back(min(intervals[i][0], res.back()[0]));
                temp.push_back(max(intervals[i][1], res.back()[1]));
                res.erase(res.end() - 1);
                res.push_back(temp);
                // res.back()[1] = max(res.back()[1], intervals[i][1]);  // 仅更新“新区间”的结束坐标就可以（上面是之前自己的实现，太麻烦，不推荐）
            }
        }
        
        return res;
    }
    
    // 这里要传引用，加快排序速度（重要！否则是值传递，很慢）
    static bool cmp(vector<int> &A, vector<int> &B){
        return A[0] < B[0];
    }
    
};