(合併 重疊陣列)練習容器的用法

Kobe10發表於2017-02-05
  • 題目
    給定間隔的集合,合併所有重疊間隔。
    
    例如,
    給定[1,3],[2,6],[8,10],[15,18]
    return [1,6],[8,10],[15,18]。

     

  • 題目分析:合併重疊的陣列,這裡首先是定義排序規則,給所有陣列進行排序,然後合併重疊的就行了。
  • 程式碼
    /**
     * 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 comp(const Interval &a, const Interval &b){
            if (a.start == b.start)//如果第一個元素相同,就按照最後一個元素比較;如果不同就按第一個元素來排序
                return a.end < b.end;
            else
                return a.start < b.start;
        }
    class Solution {
    public:
        vector<Interval> merge(vector<Interval> &intervals) {
            sort(intervals.begin(), intervals.end(), comp);//定義排序規則,按照升序排序
            vector<Interval> res;
            int len = intervals.size();
            for (int i=0; i<len; i++){
                if (res.empty())
                    res.push_back(intervals[i]);
                else{
                    Interval last = res.back();//找到res裡面的最後一個元素與當前interval元素進行比較
                    if (last.end >= intervals[i].start){//當前一個元素end大於當前元素start時,在比較end與end之間,進行合併
                        res.pop_back();
                        last.end = max(last.end, intervals[i].end);
                        res.push_back(last);
                    }else//如果不能合併,直接入容器
                        res.push_back(intervals[i]);
                }
            }
            return res;
        }
    
    };

     

相關文章