leetcode352数据流拆分

题目信息

给你一个由非负整数 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]]

题目理解

不相交区间，说明相邻的区间需要合并，即1，2应该合并为一个区间[1,2]

方法一：双指针：快慢指针

class SummaryRanges {
    static List<Integer> list;
    
    public SummaryRanges() {
        list = new ArrayList<>();
    }
    
    public void addNum(int val) {
        list.add(val);
    }

    /**
     *快慢指针
     * @return
     */
    public int[][] getIntervals() {
        list.sort(Comparator.comparingInt(a -> a));
        List<int[]> res = new ArrayList<>();
        int pre = -1;
        int cur = -1;
        for (Integer x : list) {
        	//pre + 1 != x 判断是否连续，即1，2，3这种情况
        	//pre != x 判断是否重复，即1，1，1这种情况
        	//pre != -1 是为了将pre赋值
            if (pre != -1 && pre + 1 != x  && pre != x) {
                res.add(new int[]{cur, pre});
                cur = x;
            }
            if (cur == -1) {
                cur = x;
            }
            pre = x;
        }
        //将最后情况添加进去
        res.add(new int[]{cur, pre});
        return res.toArray(new int[0][]);
    }
}

方法二，treeMap

方法一，每次都要排序，在遍历，有没有方法能够避免排序？方法就是在插入时实现排序，使用treeMap记录排序区间信息，转换为图论问题即可。

情况一：

[l,r]完全包裹val，则直接跳过

情况二：

[l,r]区间满足 r + 1 = val，则删除原区间，并增加新区间[l,val]

情况三：

[l,r]区间满足 l - 1 = val，则删除原区间，并增加新区间[val,r]

情况四：

[l0,r0],[l1,r1]两个区间满足r0 + 1 = l1 - 1 = val，则删除原来两个区间，增加新区间[l0,r1]，如[1,2],[4,5]，val = 3

情况五：

不满足以上四种情况，增加区间[val,val]

class SummaryRanges {
    TreeMap<Integer, Integer> preMap;

    public SummaryRanges() {
        preMap = new TreeMap<>();
    }

    public void addNum(int val) {
        //寻找l1 > val大的区间，且l1最小
        Map.Entry<Integer, Integer> entry1 = preMap.ceilingEntry(val + 1);
        //寻找l0 <= val的区间且l0最大
        Map.Entry<Integer, Integer> entry0 = preMap.floorEntry(val);
        if (entry0 != null && entry0.getKey() <= val && val <= entry0.getValue()) {
        } else {
            boolean leftFlag = entry1 != null && entry1.getKey() - 1 == val;
            boolean rightFlag = entry0 != null && entry0.getValue() + 1 == val;
            if (leftFlag && rightFlag) {
                int right = entry1.getValue();
                int left = entry0.getKey();
                preMap.remove(entry0.getKey());
                preMap.remove(entry1.getKey());
                preMap.put(left, right);
            }else if (leftFlag) {
                int right = entry1.getValue();
                preMap.remove(entry1.getKey());
                preMap.put(val, right);
            }else if (rightFlag) {
                preMap.put(entry0.getKey(), val);
            } else {
                preMap.put(val, val);
            }
        }
    }

    public int[][] getIntervals() {
        int size = preMap.size();
        int[][] ans = new int[size][2];
        int index = 0;
        for (Map.Entry<Integer, Integer> entry : preMap.entrySet()) {
            ans[index][0] = entry.getKey();
            ans[index][1] = entry.getValue();
            index++;
        }
        return ans;
    }
}