leetcode56 Merge Intervals

"""
Given a collection of intervals, merge all overlapping intervals.
Example 1:
Input: [[1,3],[2,6],[8,10],[15,18]]
Output: [[1,6],[8,10],[15,18]]
Explanation: Since intervals [1,3] and [2,6] overlaps, merge them into [1,6].
Example 2:
Input: [[1,4],[4,5]]
Output: [[1,5]]
Explanation: Intervals [1,4] and [4,5] are considered overlapping.
"""
"""
将列表按照第一个值排序。
重合情况。
拿intervals里要比较的数组右端值和已经存在res中的左端值比较，
若前者<=后者，则重叠，更新右端值并存入res。
不重合情况。将此intervals存入res即可。
"""
class Solution:
    def merge(self, intervals):
        intervals.sort(key=lambda x: x[0])   # !!!把intervals按照第一个值进行排序
        # print(sorted(C, key=lambda x: x[2]))
        # x:x[]字母可以随意修改，排序方式按照中括号[]里面的维度进行排序，
        # [0]按照第一维排序，[2]按照第三维排序
        #bug 用sorted(intervals) 带0的案例没有通过
        #Input: [[1,4],[0,4]]
        #Output: [[1,4]]
        #Expected: [[0,4]]
        n = len(intervals)
        if n == 0:
            return []
        res = [intervals[0]]
        for i in range(1, n):
            if intervals[i][0] <= res[-1][1]: #重合。
                # !!!res[-1][1] 第一个-1指的是索引从后往前第一个，用-1表示
                res[-1][1] = max(res[-1][1], intervals[i][1])
            else:#不重合
                res.append(intervals[i])
        return res