56. 合并区间

发布时间 2023-05-10 11:09:25作者: xiazichengxi

以数组 intervals 表示若干个区间的集合,其中单个区间为 intervals[i] = [starti, endi] 。请你合并所有重叠的区间,并返回 一个不重叠的区间数组,该数组需恰好覆盖输入中的所有区间 。


输入:intervals = [[1,3],[2,6],[8,10],[15,18]]
输出:[[1,6],[8,10],[15,18]]
解释:区间 [1,3] 和 [2,6] 重叠, 将它们合并为 [1,6].

我的解法


class Solution {
public:
    vector<vector<int>> merge(vector<vector<int>>& intervals) {
        int len = intervals.size();
        if(len ==  1) return intervals;
        sort(intervals.begin(),intervals.end(),[](vector<int> &lhs,vector<int> &rhs)->bool{
            if(lhs[0] != rhs[0]){
                return lhs[0] < rhs[0];
            }
            else{
                return lhs[1] < rhs[1];
            }
        });
        vector<vector<int>> res;
        int start = intervals[0][0];
        int end = intervals[0][1];
        for(int i = 1;i < len; i++){
            if(intervals[i][0] > end){
                res.push_back({start,end});
                start = intervals[i][0];
                end = intervals[i][1];
                
            }
            else{
                end = std::max(intervals[i][1],end);
            }
            if(i == len - 1){
                res.push_back({start,end});
            }
        }
        return res;
    }
};

标准解法


class Solution {
public:
    vector<vector<int>> merge(vector<vector<int>>& intervals) {
        vector<vector<int>> result;
        if (intervals.size() == 0) return result; // 区间集合为空直接返回
        // 排序的参数使用了lambda表达式
        sort(intervals.begin(), intervals.end(), [](const vector<int>& a, const vector<int>& b){return a[0] < b[0];});

        // 第一个区间就可以放进结果集里,后面如果重叠,在result上直接合并
        result.push_back(intervals[0]); 

        for (int i = 1; i < intervals.size(); i++) {
            if (result.back()[1] >= intervals[i][0]) { // 发现重叠区间
                // 合并区间,只更新右边界就好,因为result.back()的左边界一定是最小值,因为我们按照左边界排序的
                result.back()[1] = max(result.back()[1], intervals[i][1]); 
            } else {
                result.push_back(intervals[i]); // 区间不重叠 
            }
        }
        return result;
    }
};