题目地址:
https://www.lintcode.com/problem/merge-intervals/description
给定一列区间(不一定排好序),要求返回所有区间合并后的结果。
思路是,先对区间排序,然后开始一个一个加进最终结果res中。每次加进去的时候就与res最后一个区间进行合并。代码如下:
import java.util.ArrayList;
import java.util.Comparator;
import java.util.List;
public class Solution {
/**
* @param intervals: interval list.
* @return: A new interval list.
*/
public List<Interval> merge(List<Interval> intervals) {
// write your code here
List<Interval> res = new ArrayList<>();
if (intervals == null || intervals.isEmpty()) {
return res;
}
// 先对区间进行排序,左端点小者在前面
intervals.sort(Comparator.comparingInt(i -> i.start));
// 先把第一个区间加进res
res.add(intervals.get(0));
for (int i = 1; i < intervals.size(); i++) {
// 每遍历到一个区间,就将这个区间与res末尾的区间进行合并
mergeIn(res, res.get(res.size() - 1), intervals.get(i));
}
return res;
}
private void mergeIn(List<Interval> res, Interval last, Interval cur) {
// 如果两个区间不重合,则直接把当前区间加进res;否则就拓宽res最后一个区间的右边界
if (last.end < cur.start) {
res.add(cur);
} else {
last.end = Math.max(last.end, cur.end);
}
}
}
class Interval {
public int start, end;
Interval(int start, int end) {
this.start = start;
this.end = end;
}
}
时间复杂度 O ( n log n ) O(n\log n) O(nlogn),空间 O ( 1 ) O(1) O(1)。