LeetCode 56.合并区间

LeetCode 56. 合并区间

以数组 intervals 表示若干个区间的集合，其中单个区间为 intervals[i] = [starti, endi] 。请你合并所有重叠的区间，并返回一个不重叠的区间数组，该数组需恰好覆盖输入中的所有区间。

示例 1：

输入：intervals = [[1,3],[2,6],[8,10],[15,18]]
输出：[[1,6],[8,10],[15,18]]
解释：区间 [1,3] 和 [2,6] 重叠, 将它们合并为 [1,6].

示例 2：

输入：intervals = [[1,4],[4,5]]
输出：[[1,5]]
解释：区间 [1,4] 和 [4,5] 可被视为重叠区间。

题解

class Solution {
    /**
    思路：先将每一个区间的上限数和下限数独立出来，然后进行排序，需要标记这个数是区间的上限还是下限，然后遍历这些数，如果是下限就+1，上限就-1，减到0时说明是一个新的大区间，加入到结果集里
     */
    static class Entry implements Comparable{
        int num;
        int mark; //0表示是节点的头部，1表示是节点的尾部

        public Entry(int num,int mark){
            this.num=num;
            this.mark=mark;
        }

        public int compareTo(Object o){
            return this.num-((Entry)o).num;
        }
    }

    public int[][] merge(int[][] intervals) {
        List<Entry> list=new ArrayList<>();
        for(int i=0;i<intervals.length;i++){
            list.add(new Entry(intervals[i][0],0));
            list.add(new Entry(intervals[i][1],1));
        }
        Entry[] collect= list.stream().sorted().toArray(Entry[]::new);
        int[] temp=new int[2];
        int remark=0; //记录出现上限下限的次数
        List<int[]> res=new ArrayList<>();
        for(int i=0;i<collect.length;i++){
            if(collect[i].mark==0){
                remark++;
                if(i>0&&collect[i].num==collect[i-1].num&&collect[i-1].mark==1){
                    continue;
                }
                if(remark==1){
                    temp[0]=collect[i].num;
                }
            }
            else{
                remark--;
                if(i<collect.length-1&&collect[i].num==collect[i+1].num&&collect[i+1].mark==0){
                    continue;
                }
                if(remark==0){
                    temp[1]=collect[i].num;
                    res.add(new int[]{temp[0],temp[1]});
                }
            }
        }
        return res.toArray(new int[0][0]);
    }

}