/*
* @lc app=leetcode id=56 lang=javascript
*
* [56] Merge Intervals
*/
/**
* @param {number[][]} intervals
* @return {number[][]}
*/
function intersected(a, b) {
if (a[0] > b[1] || a[1] < b[0]) return false;
return true;
}
function mergeTwo(a, b) {
return [Math.min(a[0], b[0]), Math.max(a[1], b[1])];
}
var merge = function (intervals) {
// 这种算法需要先排序
intervals.sort((a, b) => a[0] - b[0]);
for (let i = 0; i < intervals.length - 1; i++) {
const cur = intervals[i];
const next = intervals[i + 1];
if (intersected(cur, next)) {
intervals[i] = undefined;
intervals[i + 1] = mergeTwo(cur, next);
}
}
return intervals.filter((q) => q);
};
Python3 Code:
class Solution:
def merge(self, intervals: List[List[int]]) -> List[List[int]]:
"""先排序，后合并"""
if len(intervals) <= 1:
return intervals
# 排序
def get_first(a_list):
return a_list[0]
intervals.sort(key=get_first)
# 合并
res = [intervals[0]]
for i in range(1, len(intervals)):
if intervals[i][0] <= res[-1][1]:
res[-1] = [res[-1][0], max(res[-1][1], intervals[i][1])]
else:
res.append(intervals[i])
return res
复杂度分析
令 n 为 intervals 的长度。
时间复杂度：由于采用了排序，因此复杂度大概为 $O(nlogn)$
空间复杂度：$O(n)$
扩展
参考 Python3 代码：
class Solution:
def solve(self, intervals, cut):
ans = []
for s, e in intervals:
if s < cut[0]: ans.append([s, min(e, cut[0])])
if cut[1] < e: ans.append([max(s, cut[1]), e])
return ans
