-
Notifications
You must be signed in to change notification settings - Fork 0
/
InsertInterval.java
37 lines (37 loc) · 1.57 KB
/
InsertInterval.java
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
public class Solution {
public List<Interval> insert(List<Interval> intervals, Interval newInterval) {
intervals.add(newInterval);
intervals.sort((a,b)->Integer.compare(a.start, b.start));
List<Interval> res = new LinkedList<>();
res.add(intervals.get(0));
int size = intervals.size();
for(int i=1;i<size;i++){
Interval cur = intervals.get(i);
Interval pre = res.get(res.size()-1);
if(cur.start<=pre.end){
pre.end = pre.end>=cur.end? pre.end:cur.end;
}else{
res.add(cur);
}
}
return res;
}
public List<Interval> insert2(List<Interval> intervals, Interval newInterval) {
List<Interval> result = new LinkedList<>();
int i = 0;
// add all the intervals ending before newInterval starts
while (i < intervals.size() && intervals.get(i).end < newInterval.start)
result.add(intervals.get(i++));
// merge all overlapping intervals to one considering newInterval
while (i < intervals.size() && intervals.get(i).start <= newInterval.end) {
newInterval = new Interval( // we could mutate newInterval here also
Math.min(newInterval.start, intervals.get(i).start),
Math.max(newInterval.end, intervals.get(i).end));
i++;
}
result.add(newInterval); // add the union of intervals we got
// add all the rest
while (i < intervals.size()) result.add(intervals.get(i++));
return result;
}
}