leetcode56题:合并区间

leetcode56题:合并区间

  • 给出一个区间的集合, 请合并所有重叠的区间。
    示例:
    给出 [1,3],[2,6],[8,10],[15,18],
    返回 [1,6],[8,10],[15,18]

  • 思路:先按照各个区间的第1个元素进行排序(这里采用快速排序),要注意的是,当第1个元素相等时,按照第2个元素进行排序,最终得到排好序的区间sort_list. 然后定义2个指针start和end,分别定义区间的第1个元素和第2个元素,即其min和max.当end<下1个区间的start(即min),则说明当前[start,end]和下一区间没有重叠,即直接加入结果列表res中,然后重新赋值start=下一区间的min,end=下一区间的max,再判断下个区间是不是最后1个区间,如果是的话,直接加入结果列表res;
    当end>=下一区间的start(即min)时,则说明当前[start,end]和下一区间有重叠,则2个区间需要合并,因此需要更新这个大区间的max,所以end = max[end, 下一区间的max],接着再判断下一区间是否是最后1个区间,如果是,则直接res.append([start,end]),如果不是最后1个区间,则end需要和下下区间的min比较,如果end<下下区间的min,则当前这个大区间合并完成,res.append([start,end])后,更新start和end,如果end>=下下区间的min,则没有合并完成,continue.`
# Definition for an interval.
# class Interval:
#     def __init__(self, s=0, e=0):
#         self.start = s
#         self.end = e

class Solution:
    def merge(self, intervals):
        """
        :type intervals: List[Interval]
        :rtype: List[Interval]
        """
        #先对所有区间的第1个元素进行排序
        n = len(intervals)
        if n<2:
            return intervals
        sort_list = self.quick_sort_list(intervals)
        start = sort_list[0].start
        end = sort_list[0].end
        res = []
        for i in range(1,n):
            if end<sort_list[i].start:
                res.append([start,end])
                start = sort_list[i].start
                end = sort_list[i].end
                if i+1==n:
                    res.append([start,end])
            elif end>=sort_list[i].start:
                if end<sort_list[i].end:
                    end = sort_list[i].end
                if i+1<n:
                    if end<sort_list[i+1].start:
                        res.append([start,end])
                        start = sort_list[i+1].start
                        end = sort_list[i+1].end
                else:
                    res.append([start,end])
        return res

    def quick_sort_list(self,intervals):
        less = []
        more = []
        pivotlist = []
        if len(intervals)<=1:
            return intervals
        else:
            pivot = intervals[0]
            for i in range(len(intervals)):
                item_list = intervals[i]
                if item_list.start<pivot.start:
                    less.append(item_list)
                elif item_list.start>pivot.start:
                    more.append(item_list)
                else:
                    if item_list.end<pivot.end:
                        less.append(item_list)
                    elif item_list.end>pivot.end:
                        more.append(item_list)
                    else:
                        pivotlist.append(pivot)
        less = self.quick_sort_list(less)
        more = self.quick_sort_list(more)
        return less + pivotlist + more

猜你喜欢

转载自blog.csdn.net/qq_35559420/article/details/79809092