0352：将数据流变为多个不相交区间（★★）

2015-12-18 约 412 字预计阅读 1 分钟次阅读

力扣第 352 题

题目

给你一个由非负整数 a₁, a₂, ..., a_n 组成的数据流输入，请你将到目前为止看到的数字总结为不相交的区间列表。

实现 SummaryRanges 类：

SummaryRanges() 使用一个空数据流初始化对象。
void addNum(int val) 向数据流中加入整数 val 。
int[][] getIntervals() 以不相交区间 [start_i, end_i] 的列表形式返回对数据流中整数的总结。

示例：

输入：
["SummaryRanges", "addNum", "getIntervals", "addNum", "getIntervals", "addNum", "getIntervals", "addNum", "getIntervals", "addNum", "getIntervals"]
[[], [1], [], [3], [], [7], [], [2], [], [6], []]
输出：
[null, null, [[1, 1]], null, [[1, 1], [3, 3]], null, [[1, 1], [3, 3], [7, 7]], null, [[1, 3], [7, 7]], null, [[1, 3], [6, 7]]]

解释：
SummaryRanges summaryRanges = new SummaryRanges();
summaryRanges.addNum(1);      // arr = [1]
summaryRanges.getIntervals(); // 返回 [[1, 1]]
summaryRanges.addNum(3);      // arr = [1, 3]
summaryRanges.getIntervals(); // 返回 [[1, 1], [3, 3]]
summaryRanges.addNum(7);      // arr = [1, 3, 7]
summaryRanges.getIntervals(); // 返回 [[1, 1], [3, 3], [7, 7]]
summaryRanges.addNum(2);      // arr = [1, 2, 3, 7]
summaryRanges.getIntervals(); // 返回 [[1, 3], [7, 7]]
summaryRanges.addNum(6);      // arr = [1, 2, 3, 6, 7]
summaryRanges.getIntervals(); // 返回 [[1, 3], [6, 7]]

提示：

0 <= val <= 10⁴
最多调用 addNum 和 getIntervals 方法 3 * 10⁴ 次

进阶：如果存在大量合并，并且与数据流的大小相比，不相交区间的数量很小，该怎么办?

分析

0057 升级版，可以用数组，也可以用有序集合维护区间。

解答

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18


class SummaryRanges:

    def __init__(self):
        from sortedcontainers import SortedList
        self.sl = SortedList()

    def addNum(self, value: int) -> None:
        pos = self.sl.bisect_left((value+1,))
        l,r = value,value
        if pos<len(self.sl) and self.sl[pos][0]==value+1:
            r = self.sl.pop(pos)[1]
        if pos and self.sl[pos-1][1]>=value-1:
            a,b = self.sl.pop(pos-1)
            l,r = a,max(r,b)
        self.sl.add((l,r))

    def getIntervals(self) -> List[List[int]]:
        return list(self.sl)

43 ms

目录

0352：将数据流变为多个不相交区间（★★）

题目

分析

解答