[LeetCode] 56. Merge Intervals

思路:
先对数据数组进行排序. 然后用一个整数记录当前区间的开始端, 另一个做结束端, 如果后面的区间开始端小于等于当前的结束端, 那么就要把他们merge到一起, 这时候前端不用考虑, 当前全局后端和遍历到的当前后端的最大值, 会被更新为全局后端. 然后如果遍历到的区间前端大于之前的全局后端, 那么之前的前后端就形成了一个区间, 我们把它们放到结果数组内, 最后别忘了最后一组全局前端后端没有区间能够替代他们了, 所以他们在for循环内就没有机会被push到结果中, 所以在循环结束后再push一次, 这样结果才完整.

class IntervalCmp {
public:
    bool operator()(Interval it1, Interval it2) {
        if (it1.start == it2.start)
            return it1.end < it2.end;
        return it1.start < it2.start;
    }
};

vector<Interval> merge(vector<Interval>& intervals) {
    if (intervals.size() == 1 || intervals.empty())
        return intervals;
    sort(intervals.begin(), intervals.end(), IntervalCmp());
    vector<Interval> res;

    int curStart = intervals[0].start, curEnd = intervals[0].end;
    for (int i = 0; i < intervals.size(); i++) {
        if (intervals[i].start <= curEnd)
            curEnd = max(curEnd, intervals[i].end);
        else {
            res.push_back(Interval(curStart, curEnd));
            curStart = intervals[i].start;
            curEnd = intervals[i].end;
        }
    }
    res.push_back(Interval(curStart, curEnd));

    return res;
}