【LeetCode】56. 合并区间 (C++)

原题地址:https://leetcode-cn.com/problems/merge-intervals/description/

题目描述:

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

示例 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] 可被视为重叠区间。

解题方案:

这道题最核心的部分是排序,开始写的cmp里面写成了<=,然后提交会出错,想了很久都没发现这个问题,看了答案之后发现这样会出现乱序的问题。

代码:

/**
 * Definition for an interval.
 * struct Interval {
 *     int start;
 *     int end;
 *     Interval() : start(0), end(0) {}
 *     Interval(int s, int e) : start(s), end(e) {}
 * };
 */
    bool cmp(Interval a, Interval b)
    {
        return a.start < b.start;
    }
class Solution {

public:
    vector merge(vector& intervals) {
        vector out;
        if(intervals.size() == 0)
            return out;
        sort(intervals.begin(), intervals.end(), cmp);
        int i = 1;
        out.push_back(intervals[0]);
        int count = 0;
        while(i < intervals.size())
        {
            if(intervals[i].start > out[count].end)
            {
                out.push_back(intervals[i]);
                i ++;
                count ++;
            }
            else
            {
                while(intervals[i].start <= out[count].end && i < intervals.size())
                {
                    int e = intervals[i].end > out[count].end ? intervals[i].end : out[count].end;
                    Interval in(out[count].start, e);
                    out.erase(out.end());
                    out.push_back(in);
                    i ++;
                }
                // count ++;
            }
        }
        return out;
    }
};

 

你可能感兴趣的:(Leetcode)