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

示例 3：

输入：intervals = [], newInterval = [5,7]
输出：[[5,7]]

示例 4：

输入：intervals = [[1,5]], newInterval = [2,3]
输出：[[1,5]]

示例 5：

输入：intervals = [[1,5]], newInterval = [2,7]
输出：[[1,7]]

提示：

0 <= intervals.length <= 10⁴
intervals[i].length == 2
0 <= intervals[i][0] <= intervals[i][1] <= 10⁵
intervals 根据 intervals[i][0] 按升序排列
newInterval.length == 2
0 <= newInterval[0] <= newInterval[1] <= 10⁵

解法一：一次遍历

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43


func min(nums ...int)int {
    res := nums[0]
    for _, num := range nums {
        if num < res {
            res = num
        }
    }
    return res
}

func insert(intervals [][]int, newInterval []int) [][]int {
    left, right := newInterval[0], newInterval[1]
    n := len(intervals)
    j := 0
    var ans [][]int
    for j < n && intervals[j][1] < left {
        ans = append(ans, intervals[j])
        j++
    }
    if j == n {
        ans = append(ans, newInterval)
    } else {
        start := min(intervals[j][0], left)
        for j < n && intervals[j][1] <= right {
            j++
        }
        if j == n {
            ans = append(ans, []int{start, right})
        } else {
            if intervals[j][0] > right {
                ans = append(ans, []int{start, right})
            } else {
                ans = append(ans, []int{start, intervals[j][1]})
                j++
            }
        }
    }
    for j < n {
        ans = append(ans, intervals[j])
        j++
    }
    return ans
}

官方题解如下：

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39


func insert(intervals [][]int, newInterval []int) (ans [][]int) {
    left, right := newInterval[0], newInterval[1]
    merged := false
    for _, interval := range intervals {
        if interval[0] > right {
            // 在插入区间的右侧且无交集
            if !merged {
                ans = append(ans, []int{left, right})
                merged = true
            }
            ans = append(ans, interval)
        } else if interval[1] < left {
            // 在插入区间的左侧且无交集
            ans = append(ans, interval)
        } else {
            // 与插入区间有交集，计算它们的并集
            left = min(left, interval[0])
            right = max(right, interval[1])
        }
    }
    if !merged {
        ans = append(ans, []int{left, right})
    }
    return
}

func min(a, b int) int {
    if a < b {
        return a
    }
    return b
}

func max(a, b int) int {
    if a > b {
        return a
    }
    return b
}

总结：对于两个有重叠的区间 [l1, r1], [l2, r2]，合并后的区间为 [min(l1, l2), max(r1, r2)]。