【Golang】LeetCode 56. 合并区间

56. 合并区间

题目描述

思路

解决这道问题我们首先需要对intervals数组进行一道排序。

在 Golang 当中借助库函数sort的Slice方法我们可以实现对指定序列按照自定义规则的排序；在 Python 当中我们通过sorted函数对数组进行外部排序（返回值是排序后的数组）或是直接使用sort方法来对数组进行原地排序。

完成排序之后，剩下要做的工作就非常简单了，我们只需要设置两个变量l和r来标定当前的连续区间即可。具体来说，我们令l, r = intervals[0][0], intervals[0][1]，然后从下标1开始对整个数组进行遍历。

遍历的过程当中，每一次我们只需要判断r和当前所遍历区间左端点intervals[i][0]之间的大小关系即可，如果r < intervals[i][0]，则说明找到了一个新的区间（因为当前区间[l, r]与这个区间不重叠），此时我们记录[l, r]到答案当中，并更新l, r = intervals[i][0], intervals[i][1]即可。

反之，如果r >= intervals[i][0]，说明当前区间与我们正在维护的区间存在重叠，此时令r = max(r, intervals[i][1])即可，原因是当前区间的右端点可能比我们维护区间的右端点更长，这两个区间已经被合并，应该一道更新右端点的位置。

基于以上思路，我们写代码来解决问题。

Golang 题解

go 复制代码

func merge(intervals [][]int) [][]int {
    sort.Slice(intervals, func (i, j int) bool {
        return intervals[i][0] < intervals[j][0]
    })
    ans := [][]int{}
    l, r := intervals[0][0], intervals[0][1]
    for i := 1; i < len(intervals); i ++ {
        if r < intervals[i][0] {
            ans = append(ans, []int{l, r})
            l, r = intervals[i][0], intervals[i][1]
        } else {
            r = max(r, intervals[i][1])
        }
    }
    ans = append(ans, []int{l, r})
    return ans
}

Python 题解

python 复制代码

class Solution:
    def merge(self, intervals: List[List[int]]) -> List[List[int]]:
        intervals = sorted(intervals, key=lambda x: (x[0], x[1]))
        ans, l, r = [], intervals[0][0], intervals[0][1]

        for i in range(1, len(intervals)):
            if r < intervals[i][0]:
                ans.append([l, r])
                l, r = intervals[i][0], intervals[i][1]
            else:
                r = max(r, intervals[i][1])
        ans.append([l, r])
        return ans