LeetCode - 解题笔记 - 56 - Merge Intervals

Merge Intervals

Solution 1

这个题的合并判定可以简化到一个情形：保证输入合法的前提下，前一个区间的结尾不小于后一个区间的开头时，即可进行合并。

所以没必要逐个考察所有区间，只需要保证上述判定过程不会遗漏即可，而这一原则可以直接通过对区间排序实现：优先开头位置，然后是结束位置。这样能够保证存在合并可能的区间必然是连续的。所以线性遍历一次就可以保证不会出现遗漏了。

但是！LeetCode的编译器好像有bug，我使用自定义的比较函数的情况下会出现null pointer然后报错（本地编译器不会出现这个问题，输入也确实没有空项，空项实在排序过程中出现的）。所以只能使用默认的比较策略（默认逻辑上是接近的，但是相等策略下有略微的不同）了，为了保证思路准确，我把自定义函数留下了。

• 时间复杂度：$O(n\log n)$，其中$n$为输入区间的个数，排序的理想时间下界，后续操作都是常数复杂度
• 空间复杂度：$O(n)$，其中$n$为输入区间的个数，排序时的临时空间占用

class Solution {
public:
    vector<vector<int>> merge(vector<vector<int>>& intervals) {
        vector<vector<int>> ans;
        
        for (auto interval: intervals) {
            if (interval.size() < 2) { cout << "WWWW" << endl;}
        }
        
        sort(intervals.begin(), intervals.end());
        // sort(intervals.begin(), intervals.end()， compare); // runtime error
        
        for (auto interval: intervals) {
            if (ans.empty() || ans.back()[1] < interval[0]) {
                ans.push_back(interval);
            }
            else {
                ans.back()[1] = max(ans.back()[1], interval[1]);
            }
        }
        
        return ans;
    }
private:
    static bool compare(const vector<int>& a, const vector<int>& b)
    {
        if (a[0] < b[0]) {
            return true;
        }
        else if (a[0] == b[0] && a[1] <= b[1]) {
            return true;
        }
        return false;
    }
};

Solution 2

Solution 1的Python实现，又学习到了一个python排序的小知识

class Solution:
    def merge(self, intervals: List[List[int]]) -> List[List[int]]:
        
        def compare(a, b):
            if a[0] < b[0]: return -1
            elif a[0] == b[0] and a[1] < b[1]: return -1
            elif a[0] == b[0] and a[1] == b[1]: return 0
            else: return 1
            
        ans = list()
            
        intervals = sorted(intervals, key=functools.cmp_to_key(compare))
        
        for interval in intervals:
            if len(ans) == 0 or ans[-1][1] < interval[0]:
                ans.append(interval)
            else:
                ans[-1][1] = max(ans[-1][1], interval[1])

        return ans