给你一个由非负整数 a1, a2, ..., an
组成的数据流输入,请你将到目前为止看到的数字总结为不相交的区间列表。
实现 SummaryRanges
类:
SummaryRanges()
使用一个空数据流初始化对象。void addNum(int val)
向数据流中加入整数val
。int[][] getIntervals()
以不相交区间[starti, endi]
的列表形式返回对数据流中整数的总结。
输入: ["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 <= value <= 104
- 最多调用
addNum
和getIntervals
方法3 * 104
次
**进阶:**如果存在大量合并,并且与数据流的大小相比,不相交区间的数量很小,该怎么办?
from sortedcontainers import SortedList
class SummaryRanges:
def __init__(self):
self.intervals = SortedList()
self.nums = set()
def addNum(self, value: int) -> None:
if value in self.nums:
return
i = self.intervals.bisect_left([value, value])
self.nums.add(value)
if value - 1 in self.nums and value + 1 in self.nums:
x, _ = self.intervals.pop(i - 1)
_, y = self.intervals.pop(i - 1)
self.intervals.add([x, y])
elif value - 1 in self.nums:
x, _ = self.intervals.pop(i - 1)
self.intervals.add([x, value])
elif value + 1 in self.nums:
_, y = self.intervals.pop(i)
self.intervals.add([value, y])
else:
self.intervals.add([value, value])
def getIntervals(self) -> List[List[int]]:
return list(self.intervals)
# Your SummaryRanges object will be instantiated and called as such:
# obj = SummaryRanges()
# obj.addNum(value)
# param_2 = obj.getIntervals()