56. Merge Intervals

56. Merge Intervals

Medium

Given a collection of intervals, merge all overlapping intervals.

Example 1:

Input: [[1,3],[2,6],[8,10],[15,18]]
Output: [[1,6],[8,10],[15,18]]
Explanation: Since intervals [1,3] and [2,6] overlaps, merge them into [1,6].

Example 2:

Input: [[1,4],[4,5]]
Output: [[1,5]]
Explanation: Intervals [1,4] and [4,5] are considered overlapping.

/**
 * Definition for an interval.
 * public class Interval {
 *     int start;
 *     int end;
 *     Interval() { start = 0; end = 0; }
 *     Interval(int s, int e) { start = s; end = e; }
 * }
 */
class Solution {
    public List<Interval> merge(List<Interval> intervals) {
        if (intervals.isEmpty()) {
            return intervals;
        }
        
        Collections.sort(intervals, (o1, o2) -> {
            Interval i1 = (Interval) o1;
            Interval i2 = (Interval) o2;
            
            if (Integer.compare(i1.start, i2.start) == 0) {
                return Integer.compare(i1.end, i2.end);
            } else {
                return Integer.compare(i1.start, i2.start);
            }
        });
        
        Stack<Interval> stack = new Stack<>();
        stack.push(intervals.get(0));
        
        for (int i = 1; i < intervals.size(); i++) {
            Interval ii = intervals.get(i);
            Interval si = stack.pop();
            
            if (si.end >= ii.end) {
                stack.push(si);
            } else if (si.end >= ii.start) {
                stack.push(new Interval(si.start, ii.end));
            } else if (si.end < ii.start) {
                stack.push(si);
                stack.push(ii);
            }
        }
        
        return new ArrayList(stack);
    }
}