凌云的博客

行胜于言

LeetCode 算法题 56. 合并区间

分类:algorithm| 发布时间:2017-01-25 17:39:00


题目

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

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

解法

class Solution {
public:
    vector<vector<int>> merge(vector<vector<int>>& intervals) {
        if (intervals.size() <= 1) {return intervals;}
        sort(intervals.begin(), intervals.end(), [](auto &&l, auto &&r) bool { return l[0] < r[0];});

        vector<vector<int>> ret;
        vector<int> cur = intervals[0];
        for (int i = 1; i < intervals.size(); ++i) {
            if (intervals[i][0] <= cur[1]) {
                cur[1] = max(cur[1], intervals[i][1]);
            } else {
                ret.push_back(cur);
                cur = intervals[i];
            }
        }

        ret.push_back(cur);
        return ret;
    }
};