LeetCode 57. 插入区间

题目描述

给出一个无重叠的 ，按照区间起始端点排序的区间列表。
在列表中插入一个新的区间，你需要确保列表中的区间仍然有序且不重叠（如果有必要的话，可以合并区间）。

示例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] 重叠。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/insert-interval

思路解析

如果区间列表中的区间与 newInterval 无重叠，则直接插入至result；
如果区间列表中的区间与 newInterval 无重叠，则将 newInterval更新为 newInterval与当前区间的并集。

代码实现

class Solution {
private:
    vector<int> merge(vector<int> a, vector<int> b) {
        vector<int> result(2);
        result[0] = (a[0] < b[0]) ? a[0] : b[0];
        result[1] = (a[1] > b[1]) ? a[1] : b[1];
        return result;
    }
public:
    vector<vector<int>> insert(vector<vector<int>>& intervals, vector<int>& newInterval) {
        vector<vector<int>> result;
        for(auto interval : intervals) {
            if(interval[1] < newInterval[0] || interval[0] > newInterval[1])
                result.push_back(interval);
            else{
                newInterval = merge(interval, newInterval);
            }
        }
        int insertIndex = 0;
        while(insertIndex < result.size()) {
            if(newInterval[0] > result[insertIndex][0])
                insertIndex++;
            else
                break;
        }
        result.insert(result.begin() + insertIndex, newInterval);
        return result;
    }
};