LeeCode 57. 插入区间

261 阅读1分钟

题目

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

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

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

原题链接

解题思路

使用 bufferInter 作为中间变量去和数组里的数据进行比较,如果 bufferInter 的右侧,比当前范围的左侧要小,那将 bufferInter 插入,置空,然后将当前范围进行插入。

反之则进行下一步判断,当前两个范围是否有重合,我们将 tempInter 的最右侧和 bufferInter 的最左侧进行比较,如果是 >= 的话,意味着当前两个范围出现重合,需要合并两个范围,取最左最右的值作为新的 bufferInter 去进行比较,反之则没有出现重合,将当前范围添加到 result 中。

如果遍历结束后 bufferInter 依然不为空,说明还没有进行插入,此时进行插入。

代码

class Solution {
    func insert(_ intervals: [[Int]], _ newInterval: [Int]) -> [[Int]] {
        var result = [[Int]]()
        var bufferInter: [Int]? = newInterval
        intervals.forEach { (tempInter) in
            if bufferInter == nil {
                result.append(tempInter)
            } else {
                if tempInter[0] > bufferInter![1] {
                    result.append(bufferInter!)
                    bufferInter = nil
                    result.append(tempInter)
                } else {
                    if tempInter[1] >= bufferInter![0] {
                       bufferInter = [min(tempInter[0], bufferInter![0]), max(tempInter[1], bufferInter![1])]
                    } else {
                        result.append(tempInter)
                    }
                }
            }
        }
        if bufferInter != nil {
            result.append(bufferInter!)
        }
        return result
    }
}