56. 合并区间

原题

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

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

注意：输入类型已于2019年4月15日更改。 请重置默认代码定义以获取新方法签名。

提示：

$intervals[i][0]<=intervals[i][1]$

来源：力扣（LeetCode）

链接：https://leetcode-cn.com/problems/merge-intervals

著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

题解

这个题比较简单，首先按照区间的左区间进行排序，判断每个区间的右区间是否小于下一个区间的左区间，维护两个指针 $low$ 和 $high$。

如果小于：说明这两个区间可以合并 $high=max(high,intervals[i][1])$

如果大于：说明这两个区间不可以合并 $low=intervals[i][0],high=intervals[i][1]$

代码

• C++代码

class Solution {
public:
    vector<vector<int>> merge(vector<vector<int>>& intervals) {
        vector<vector<int>> ans;
        if (intervals.empty()) {
            return ans;
        }
        // c++的lambda表达式
        auto cmp = [=](const vector<int> &a, const vector<int> &b) { return a[0] < b[0]; };
        // sort函数第三个参数接受一个谓词，这个谓词可以是lambda，也可以是函数指针，也可以是函数对象（即重载()的对象）
        sort(intervals.begin(), intervals.end(), cmp);
        int size = intervals.size();
        int low = intervals[0][0], high = intervals[0][1];
        for (int i = 1; i < size; ++i) {
            if (intervals[i][0] <= high) {
                high = max(intervals[i][1], high);
            } else {
                ans.push_back({low, high});
                low = intervals[i][0];
                high = intervals[i][1];
            }
        }
        ans.push_back({low, high});
        return ans;
    }
};

• Python代码

class Solution:
    def merge(self, intervals: List[List[int]]) -> List[List[int]]:
        ans = []
        if len(intervals) == 0:
            return ans
        intervals.sort(key=lambda x : x[0])
        low, high = intervals[0][0], intervals[0][1]
        for i in range(1, len(intervals)):
            if high >= intervals[i][0]:
                high = max(high, intervals[i][1])
            else:
                ans.append([low, high])
                low = intervals[i][0]
                high = intervals[i][1]
        ans.append([low, high])
        return ans