力扣-056 合并区间

题目描述

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

样例

示例 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[i][0] <= intervals[i][1]

分析

先根据区间的左端点进行排序，然后遍历全部区间，根据区间的相交规则进行合并。
举个例子，对$intervals[i]$和$intervals[j]$判断是否相交（i<j），由于已经根据左端点进行排序，所以$intervals[i][0]\le intervals[j][0]$，因此只有当$intervals[i][1]\ge intervals[j][0]$时，区间$intervals[i]$和$intervals[j]$相交，合并成区间$[intervals[i][0],max(intervals[i][1],intervals[j][1])]$，当$intervals[i][1]<intervals[j][0]$时，区间不相交，不进行合并。
时间复杂度为$O(NlogN)$
（建议画图进行辅助理解）

代码

bool cmp(vector<int>& a, vector<int>& b) {
    return a[0] < b[0];
}

class Solution {
public:
    vector<vector<int>> merge(vector<vector<int>>& intervals) {
        vector<vector<int>> ans;
        int len = intervals.size();
        sort(intervals.begin(), intervals.end(), cmp);
        for (int i = 0; i < len; i++) {
            if (ans.size() == 0) {
                ans.push_back(intervals[i]);
            } else {
                if (ans.back()[1] < intervals[i][0]) {
                    ans.push_back(intervals[i]);
                } else {
                    ans.back()[1] = max(ans.back()[1], intervals[i][1]);
                }
            }
        }
        return ans;
    }
};