码迷,mamicode.com
首页 > 其他好文 > 详细

【Leetcode】【Hard】Insert Interval

时间:2015-08-31 23:18:19      阅读:145      评论:0      收藏:0      [点我收藏+]

标签:

Given a set of non-overlapping intervals, insert a new interval into the intervals (merge if necessary).

You may assume that the intervals were initially sorted according to their start times.

Example 1:
Given intervals [1,3],[6,9], insert and merge [2,5] in as [1,5],[6,9].

Example 2:
Given [1,2],[3,5],[6,7],[8,10],[12,16], insert and merge [4,9] in as [1,2],[3,10],[12,16].

This is because the new interval [4,9] overlaps with [3,5],[6,7],[8,10].

 

解题思路:

1、先用两次二分查找找到和待插入的区间有关系的区间范围;

2、将有关系的区间做处理;

3、生成新的合并后的区间并返回;

 

解题步骤:

 

代码:

 1 /**
 2  * Definition for an interval.
 3  * struct Interval {
 4  *     int start;
 5  *     int end;
 6  *     Interval() : start(0), end(0) {}
 7  *     Interval(int s, int e) : start(s), end(e) {}
 8  * };
 9  */
10 class Solution {
11 public:
12     static bool partial_order(const Interval &a, const Interval &b) {
13         return a.end < b.start;
14     };
15     
16     vector<Interval> insert(std::vector<Interval> &intervals, Interval newInterval) {
17         vector<Interval>::iterator less = lower_bound(intervals.begin(), intervals.end(), newInterval, partial_order);
18         vector<Interval>::iterator greater = upper_bound(intervals.begin(), intervals.end(), newInterval, partial_order);
19         
20         vector<Interval> answer; 
21         answer.insert(answer.end(), intervals.begin(), less);
22         if (less < greater) {
23             newInterval.start = min(newInterval.start, (*less).start);
24             newInterval.end = max(newInterval.end, (*(greater - 1)).end);
25         }
26         answer.push_back(newInterval);
27         answer.insert(answer.end(), greater, intervals.end());
28         
29         return answer;
30   }
31 };

 

不用lower_bound和upper_bound写的AC烂代码,留作改进:

 1 /**
 2  * Definition for an interval.
 3  * struct Interval {
 4  *     int start;
 5  *     int end;
 6  *     Interval() : start(0), end(0) {}
 7  *     Interval(int s, int e) : start(s), end(e) {}
 8  * };
 9  */
10 class Solution {
11 public:
12     vector<Interval> insert(vector<Interval>& intervals, Interval newInterval) {
13         if (intervals.empty())
14             return vector<Interval> (1, newInterval);
15         int left = 0;
16         int right = intervals.size() - 1;
17         int size = intervals.size();
18         int ins_start, ins_end;
19         
20         while (left < right) {
21             int mid = (left + right) / 2;
22             if (intervals[mid].end < newInterval.start)
23                 left = mid + 1;
24             else 
25                 right = mid;
26         }
27         ins_start = left;
28         
29         left = left == 0 ? 0 : left - 1;
30         right = intervals.size() - 1;
31         while (left < right) {
32             int mid = (left + right) / 2 + 1;
33             if (newInterval.end < intervals[mid].start)
34                 right = mid - 1;
35             else
36                 left = mid;
37         }
38         ins_end = right;
39         
40         if (ins_end == 0 && newInterval.end < intervals[0].start) {
41             intervals.insert(intervals.begin(), newInterval);
42             return intervals;
43         }
44         if (ins_start == size - 1 && newInterval.start > intervals[size - 1].end) {
45             intervals.push_back(newInterval);
46             return intervals;
47         }
48         
49         vector<Interval> answer;
50         answer.insert(answer.end(), intervals.begin(), intervals.begin() + ins_start);
51         if (ins_start <= ins_end) {
52             newInterval.start = min(newInterval.start, intervals[ins_start].start);
53             newInterval.end = max(newInterval.end, intervals[ins_end].end);
54         }
55         answer.push_back(newInterval);
56         answer.insert(answer.end(), intervals.begin() + ins_end + 1, intervals.end());
57         return answer;
58         
59     }
60 };

 

【Leetcode】【Hard】Insert Interval

标签:

原文地址:http://www.cnblogs.com/huxiao-tee/p/4774315.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!