Java实现 LeetCode 57.插入区间

给出一个无重叠的 ，按照区间起始端点排序的区间列表。

在列表中插入一个新的区间，你需要确保列表中的区间仍然有序且不重叠（如果有必要的话，可以合并区间）。

示例 1：

输入：intervals = [[1,3],[6,9]], newInterval = [2,5]
输出：[[1,5],[6,9]]
示例 2：

输入：intervals = [[1,2],[3,5],[6,7],[8,10],[12,16]], newInterval = [4,8]
输出：[[1,2],[3,10],[12,16]]
解释：这是因为新的区间 [4,8] 与 [3,5],[6,7],[8,10] 重叠。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/insert-interval
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

class Solution {
    public int[][] insert(int[][] intervals, int[] newInterval) {
        if(intervals.length == 0) {
            int[][] ans = new int[1][2];
            ans[0] = newInterval;
            return ans;
        }
        int[][] ans = new int[intervals.length + 1][intervals[0].length];
        int a = findSit(newInterval[0], intervals, 0);
        int b = findSit(newInterval[1], intervals, 1);
        int k = 0;
        int mark = 0;
        for (int[] p : intervals) {
            if (p[1] < a) {
                ans[k++] = p;
            } else if(mark == 0){
                ans[k++] = new int[]{a, b};
                mark++;

            }
            if (p[0] > b) {
                ans[k++] = p;
            }
        }
        if (mark == 0) {
            ans[k++] = new int[]{a, b};
        }
        return Arrays.copyOfRange(ans, 0, k);
    }

    private int findSit(int inter, int[][] intervals, int t) {
        int a = inter;
        for (int i = 0; i < intervals.length; i++) {
            if (inter >= intervals[i][0] && inter <= intervals[i][1]) {
                if (t == 1) {
                    a = intervals[i][1];
                } else {
                    a = intervals[i][0];
                }
            } else if (inter < intervals[i][0]) {
                if (i == 0) {
                    a = inter;
                } else if(intervals[i-1][1] < inter) {
                    a = inter;
                }
            }
        }
        return a;
    }
}