Python 二分分治解法： Leetcode 56- 合并区间

题目

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

思路

二分+分治+dfs，取区间中点，把各个区间分成三部分：在中点左边的，包含中点的（需要合并），在中点右边的，然后对两边再次进行如上操作，最后再把合并的中点与两边合并，返回答案。

时间复杂度:

大概是 O(nlogS) 罢，S是区间上下限宽度

问题想复杂了。。。

代码

class Solution:
    def merge(self, intervals: List[List[int]]) -> List[List[int]]:
        l_b = float("inf")
        r_b = -float("inf")
        for interval in intervals:
            if(interval[0] < l_b): l_b = interval[0]
            if(interval[1] > r_b): r_b = interval[1]

        n_intervals = self.dfs(intervals,l_b,r_b)
        return n_intervals

    def dfs(self,intervals,l_b,r_b):
        if(len(intervals) <= 1): return intervals
        mid = (l_b + r_b) // 2
        l_intervals = []
        r_intervals = []
        m_intervals = []
        for interval in intervals:
            if(interval[1] < mid):
                l_intervals.append(interval)
            elif(interval[0] > mid):
                r_intervals.append(interval)
            else:
                m_intervals.append(interval)
        
        m_interval = self.compress(m_intervals) if m_intervals else []
        l_intervals = self.dfs(l_intervals,l_b,mid)
        r_intervals = self.dfs(r_intervals,mid,r_b+1)       # 这个细节需要注意，避免死循环
        
        if(m_interval):
            pop_arr = []
            for i,l_interval in enumerate(l_intervals):
                if(m_interval[0] <= l_interval[1]):
                    pop_arr.append(i)
                    m_interval[0] = min(m_interval[0],l_interval[0])
            for i,j in enumerate(pop_arr):
                l_intervals.pop(j-i)
            pop_arr = []
            for i,r_interval in enumerate(r_intervals):
                if(m_interval[1] >= r_interval[0]):
                    pop_arr.append(i)
                    m_interval[1] = max(m_interval[1],r_interval[1])
            for i,j in enumerate(pop_arr):
                r_intervals.pop(j-i)
        if(not m_interval): return l_intervals + r_intervals
        return l_intervals + [m_interval] + r_intervals
    
    def compress(self,intervals):
        l_b = float("inf")
        r_b = -float("inf")
        for interval in intervals:
            if(interval[0] < l_b): l_b = interval[0]
            if(interval[1] > r_b): r_b = interval[1]
        return [l_b,r_b]