LeetCode-56-合并区间

题目

思路

1. 刚开始想的是设立10^4的数组，然后区间赋1，再进行汇总，写着写着觉得太麻烦了，直接合并也可以。
2. 于是考虑用HashMap存储，感觉正合适。毕竟二维数组没法动态修改。
3. 后面怎么调都不对，才发现最关键的一步是首先按照intervals[i][0]排序…学到了

代码

    public int[][] merge(int[][] intervals) {
        //先按第一个排序
        Arrays.sort(intervals, new Comparator<int[]>() {
            public int compare(int[] interval1, int[] interval2) {
                return interval1[0] - interval2[0];
            }
        });

        //直接找最近两个能不能合并就好了...先用HashMap接收
        HashMap<Integer,Integer> map = new HashMap<>();
        //先把第一个加入
        map.put(intervals[0][0],intervals[0][1]);
        //记录上一个key值
        int lastkey = intervals[0][0];
        for(int i=1;i<intervals.length;i++){
            //取出map里的最后一个 判断能不能合并
            int target0 = lastkey;
            int target1 = map.get(lastkey);
            //如果intervals[i][0] 比target1还大 或者intervals[i][1]比target[0]还小
            if(intervals[i][0]>target1){
                map.put(intervals[i][0],intervals[i][1]);
                lastkey=intervals[i][0];
                continue;
            }if(intervals[i][0]<target0){
                    map.put(intervals[i][0],Math.max(intervals[i][1],target1));
                    //删除上一个
                    map.remove(lastkey);
                    lastkey = intervals[i][0];
                    continue;
            }
            map.put(Math.min(target0,intervals[i][0]),Math.max(target1,intervals[i][1]));
            lastkey = Math.min(target0,intervals[i][0]);
        }
        //获得hashmap的keyset
        List<Integer> mapkey = new ArrayList<>(map.keySet());
        int[][] ans = new int[mapkey.size()][2];
        for(int i=0;i<ans.length;i++){
            ans[i][0]=mapkey.get(i);
            ans[i][1]=map.get(ans[i][0]);
        }
        return ans;
    }