天机阁

56. 合并区间

2022-07-30 · 2 min read
LeetCode

题目

以数组 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] 可被视为重叠区间。

提示:

  • 1 <= intervals.length <= 104
  • intervals[i].length == 2
  • 0 <= starti <= endi <= 104

题解

需要先将数组排序,然后合并区间

代码实现

func merge(intervals [][]int) [][]int {
	sort.Sort(Interval(intervals))
	var res [][]int
	x, y := intervals[0][0], intervals[0][1]
	res = append(res, []int{x, y})
	for i := 1; i < len(intervals); i++ {
		if intervals[i][0] <= y { // 区间有重叠
			x = min(intervals[i][0], x)
			y = max(intervals[i][1], y)
			res = res[:len(res)-1]
			res = append(res, []int{x, y})
		} else {
			// 不重叠
			x, y = intervals[i][0], intervals[i][1]
			res = append(res, []int{x, y})
		}
	}
	return res
}

func max(x, y int) int {
	if x > y {
		return x
	}
	return y
}

func min(x, y int) int {
	if x < y {
		return x
	}
	return y
}

type Interval [][]int

func (i Interval) Len() int {
	return len(i)
}

func (i Interval) Less(p, q int) bool {
	return i[p][0] < i[q][0]
}

func (i Interval) Swap(p, q int) {
	i[p], i[q] = i[q], i[p]
}