Rayman的绝顶之路——Leetcode每日一题打卡2

Leetcode56. 合并区间 题目：

给出一个区间的集合，请合并所有重叠的区间。

示例 1:

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

示例 2:

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

我的思路：

1. 首先给输入的二维数组按照第一个维度进行排序。
2. 将区间进行合并输出，考虑多种情况，包括：单区间、两个区间不相交、多个区间相交等

我的AC代码如下：

import java.util.*;

public class Leetcode56 {
    static int[][] merge(int[][] intervals) {
        if (intervals.length < 2) {
            return intervals;
        }
        //排序
        for (int i = 0; i < intervals.length; i++) {
            for (int j = i + 1; j < intervals.length; j++) {
                if (intervals[i][0] > intervals[j][0]) {
                    int[] a;
                    a = intervals[i];
                    intervals[i] = intervals[j];
                    intervals[j] = a;
                }
            }
        }
        
        List<int[]> list = new ArrayList<>();
        list.add(intervals[0]);

        for (int i = 1; i < intervals.length; i++) {
            int[] curInterval = intervals[i];
            // 每次新遍历到的列表与当前结果集中的最后一个区间的末尾端点进行比较
            int[] peek = list.get(list.size() - 1);
            if (curInterval[0] > peek[1]) {
                list.add(curInterval);
            } else {
                peek[1] = Math.max(curInterval[1], peek[1]);
            }
        }
        return list.toArray(new int[list.size()][]);
    }

    public static void main(String[] args) {
        int[][] a = {
                {1, 4},
                {0, 2},
                {3, 5}
        };
        int[][] res = merge(a);
        for (int i = 0; i < res.length; i++) {
            System.out.println(Arrays.toString(res[i]));
        }
    }
}

但是运行的速度略慢，所以我参考了官网上的解题思路对排序做了一下优化，改进后的代码如下：

import java.util.*;

public class Leetcode56 {
    static int[][] merge(int[][] intervals) {
        if (intervals.length < 2) {
            return intervals;
        }
        //排序
        Arrays.sort(intervals, (v1, v2) -> v1[0] - v2[0]);
        
        List<int[]> list = new ArrayList<>();
        list.add(intervals[0]);

        for (int i = 1; i < intervals.length; i++) {
            int[] curInterval = intervals[i];
            // 每次新遍历到的列表与当前结果集中的最后一个区间的末尾端点进行比较
            int[] peek = list.get(list.size() - 1);
            if (curInterval[0] > peek[1]) {
                list.add(curInterval);
            } else {
                peek[1] = Math.max(curInterval[1], peek[1]);
            }
        }
        return list.toArray(new int[list.size()][]);
    }

    public static void main(String[] args) {
        int[][] a = {
                {1, 4},
                {0, 2},
                {3, 5}
        };
        int[][] res = merge(a);
        for (int i = 0; i < res.length; i++) {
            System.out.println(Arrays.toString(res[i]));
        }
    }
}

2020.4.16打卡