凌云的博客

LeetCode 算法题 57. 插入区间

分类：algorithm| 发布时间：2017-02-24 07:49:00

---

题目

给出一个无重叠的 ，按照区间起始端点排序的区间列表。

在列表中插入一个新的区间，你需要确保列表中的区间仍然有序且不重叠（如果有必要的话，可以合并区间）。

示例 1:

输入: intervals = [[1,3],[6,9]], newInterval = [2,5]
输出: [[1,5],[6,9]]

示例 2:

输入: intervals = [[1,2],[3,5],[6,7],[8,10],[12,16]], newInterval = [4,8]
输出: [[1,2],[3,10],[12,16]]
解释: 这是因为新的区间 [4,8] 与 [3,5],[6,7],[8,10] 重叠。

解法

class Solution {
public:
    vector<vector<int>> insert(vector<vector<int>>& intervals, vector<int>& newInterval) {
        if (intervals.size() == 0) return {newInterval};
        vector<vector<int>> res;
        int i = 0;
        for (; i < intervals.size(); ++i) {
            if (intervals[i][1] < newInterval[0]) {
                res.push_back(intervals[i]);
                continue;
            }

            newInterval[0] = min(newInterval[0], intervals[i][0]);
            break;
        }

        for (; i < intervals.size(); ++i) {
            if (intervals[i][0] <= newInterval[1] && intervals[i][1] > newInterval[1]) {
                newInterval[1] = intervals[i][1];
                ++i;
                break;
            } else if (intervals[i][0] > newInterval[1]) {
                break;
            }
        }
        res.push_back(newInterval);

        for (; i < intervals.size(); ++i) res.push_back(intervals[i]);
        return res;
    }

};