小知识，大挑战！本文正在参与“程序员必备小知识”创作活动
352. 将数据流变为多个不相交区间
给你一个由非负整数 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]]
解题思路
使用优先队列维护有序的数据流，当需要获得不相交的区间时，我们可以遍历优先队列，判断有序的数据流可以组成的不相交区间
代码
class SummaryRanges {
PriorityQueue<Integer> pq = new PriorityQueue<>();
public SummaryRanges() {
}
public void addNum(int val) {
pq.add(val);
}
public int[][] getIntervals() {
int pre = pq.poll(), s = pre;
List<int[]> list = new ArrayList<>();
while (!pq.isEmpty()) {
Integer cur = pq.poll();
if (cur==pre) continue;
if (cur == pre + 1) {
pre++;
} else {
list.add(new int[]{s,pre});
s=pre=cur;
}
}
list.add(new int[]{s,pre});
int j=0;
int[][] res=new int[list.size()][2];
for (int[] ints : list) {
for (int i=ints[0];i<=ints[1];i++)
pq.add(i);
res[j++]=ints;
}
return res;
}
}
/**
* Your SummaryRanges object will be instantiated and called as such:
* SummaryRanges obj = new SummaryRanges();
* obj.addNum(val);
* int[][] param_2 = obj.getIntervals();
*/
