LeetCode 56. 合并区间（C++、python）

给出一个区间的集合，请合并所有重叠的区间。

示例 1:

输入: [[1,3],[2,6],[8,10],[15,18]]
输出: [[1,6],[8,10],[15,18]]
解释: 区间 [1,3] 和 [2,6] 重叠, 将它们合并为 [1,6].

示例 2:

输入: [[1,4],[4,5]]
输出: [[1,5]]
解释: 区间 [1,4] 和 [4,5] 可被视为重叠区间。

C++

/**
 * Definition for an interval.
 * struct Interval {
 *     int start;
 *     int end;
 *     Interval() : start(0), end(0) {}
 *     Interval(int s, int e) : start(s), end(e) {}
 * };
 */
class Solution {
public:
    static bool compare(Interval& a, Interval& b)
    {
        return a.start<b.start;
    }
    
    vector<Interval> merge(vector<Interval>& intervals) 
    {
        vector<Interval> res;
        int m=intervals.size();
        if(0==m)
        {
            return res;
        }
        sort(intervals.begin(),intervals.end(),compare);
        res.push_back(intervals[0]);
        for(int i=1;i<m;i++)
        {
            if(res.back().end>=intervals[i].start)
            {
                if(intervals[i].end>res.back().end)
                {
                    res.back().end=intervals[i].end;
                }
            }
            else
            {
                res.push_back(intervals[i]);
            }
        }
        return res;
    }
};

python

# 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]
        """
        res=[]
        m=len(intervals)
        if 0==m:
            return res
        intervals=sorted(intervals,key= lambda x:x.start)
        res.append(intervals[0])
        for i in range(1,m):
            if intervals[i].start<=res[-1].end:
                if intervals[i].end>res[-1].end:
                    res[-1].end=intervals[i].end
            else:
                res.append(intervals[i])
        return res