给你一个由非负整数 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 <= val <= 10的4次方
最多调用 addNum 和 getIntervals 方法 3 * 104 次
进阶:如果存在大量合并,并且与数据流的大小相比,不相交区间的数量很小,该怎么办?
#include
#include
#include
using namespace std;
class SummaryRanges {
public:
SummaryRanges() {
}
void addNum(int val) {
m_set.insert(val);
}
/* 暴力模拟 */
vector> getIntervals() {
vector> m_t;
set::iterator it = m_set.begin();
vector t;
int start, end;
start = end = *it;
t.push_back(start);
/* 循环计算区间 */
for (; it != m_set.end(); it++) {
start = *it;
/* 只要相邻的数相减大于1,则认为上个start到end为一个区间 */
if ((start - end) > 1) {
t.push_back(end);
m_t.push_back(t);
t.clear();
t.push_back(start);
}
end = *it;
}
/* 最后一个数没有取上的情况 */
if (t.size() == 1) {
t.push_back(end);
m_t.push_back(t);
t.clear();
}
return m_t;
}
private:
set m_set; /* 使用set保证有序 */
};