Data Stream as Disjoint Intervals

最新推荐文章于 2022-07-14 20:57:05 发布

原创最新推荐文章于 2022-07-14 20:57:05 发布 · 441 阅读

0 ·

CC 4.0 BY-SA版权

strStr 同时被 3 个专栏收录

360 篇文章

订阅专栏

基本功

78 篇文章

订阅专栏

68 篇文章

订阅专栏

本文介绍了一种使用Java TreeSet来管理区间的方法，并实现了一个名为SummaryRanges的类。该类能够添加数值并返回所有已添加数值的连续区间。通过自定义比较器确保区间按起始值排序，同时实现了区间合并。

第一次用TreeSet。。。

要懂得用这种结构

/**
 * Definition for an interval.
 * public class Interval {
 *     int start;
 *     int end;
 *     Interval() { start = 0; end = 0; }
 *     Interval(int s, int e) { start = s; end = e; }
 * }
 */
public class SummaryRanges {
TreeSet<Interval> set;
    /** Initialize your data structure here. */
    public SummaryRanges() {
        set = new TreeSet<>(new Comparator<Interval>(){
            @Override
            public int compare(Interval a, Interval b){
                return a.start - b.start;
            }
        });
    }
    
    public void addNum(int val) {
        Interval temp = new Interval(val, val);
        Interval low = set.floor(temp);
        if (low != null) {
            if (low.end >= val) {
                return;
            } else if (low.end + 1 == val) {
                temp.start = low.start;
                set.remove(low);
            }
        }
        Interval high = set.higher(temp);
        if (high != null) {
            if (temp.end + 1 == high.start) {
                temp.end = high.end;
                set.remove(high);
            }
        }
        set.add(temp);
    }
    
    public List<Interval> getIntervals() {
        Iterator<Interval> it = set.iterator();
        List<Interval> list = new LinkedList<>();
        while (it.hasNext()) {
            list.add(it.next());
        }
        return list;
    }
}

/**
 * Your SummaryRanges object will be instantiated and called as such:
 * SummaryRanges obj = new SummaryRanges();
 * obj.addNum(val);
 * List<Interval> param_2 = obj.getIntervals();
 */