[区间合并+模板] 区间合并模板(贪心+区间问题)

0. 前言

区间问题，大多属于贪心问题，要么按照左端点排序、要么右端点排序、要么双关键字排序啥的。具体问题具体分析。

1. 区间合并

思路：

• 按照区间左端点排序
• 初始化 st、ed 维护一个待合并区间的左右端点，初始化为负无穷
• 遍历所有输入区间
  • 若维护区间的右端点 ed 在当前遍历区间的左侧，那么说明这两个区间没有交集，则说明找到了新区间。在此，需要判断是否为初始化区间。
    • 如果不是初始化区间，说明 st、ed 维护的这个区间已经无法再合并了，就将其加入答案数组。
    • 若是初始化区间，则说明这是第一次遇到的区间，更新下它的 st、ed 为当前遍历的这个区间即可。
  • 否则，则有维护区间的右端点 ed 在当前遍历区间的右侧，同时，由于是根据左端点排序，则这两个区间的位置情况只能是相交、内含)，均可以区间合并，则，仅需要更新区间的右端点即可。
  • 最后需要再处理下最后一个区间，因为它可能更新了 ed 但并没有添加进答案数组中。

主要思想：

// 将所有存在交集的区间合并
void merge(vector<PII> &a) {
    vector<PII> res;    // 存储合并后的结果
    
    sort(a.begin(), a.end());
    
    int st = -2e9, ed = -2e9;   // 当前维护区间的左右端点
    
    for (auto e:a) {
        if (ed < e.first) {     // 无交集，找到了新区间
            if (st != -2e9) res.push_back({st, ed});    // 不能是初始区间
            
            st = e.first, ed = e.second;
        }
        else ed = max(ed, e.second);    
    }
    
    // 处理最后一个区间，且需要防止空区间
    if (st != -2e9) res.push_back({st, ed}); 
    
    a = res;
}

Biu

代码如下：

#include <iostream>
#include <vector>
#include <algorithm>

using namespace std;

typedef pair<int, int> PII;

const int N = 1e5+5;

int n;
vector<PII> a;

void merge(vector<PII> &a) {
    vector<PII> res;    // 存储合并后的结果
    
    sort(a.begin(), a.end());
    
    int st = -2e9, ed = -2e9;   // 当前维护区间的左右端点
    
    for (auto e:a) {
        if (ed < e.first) {     // 无交集，找到了新区间
            if (st != -2e9) res.push_back({st, ed});    // 不能是初始区间
            
            st = e.first, ed = e.second;
        }
        else ed = max(ed, e.second);    
    }
    
    if (st != -2e9) res.push_back({st, ed}); // 处理最后一个区间，且需要防止空区间
    
    a = res;
}

int main() {
    cin >> n;
    for (int i = 0; i < n; ++i) {
        int l, r;
        cin >> l >> r;
        a.push_back({l, r});
    }
    
    merge(a);
    
    cout << a.size() << endl;
    
    return 0;
}

在这个写法中，一开始是从一个空区间开始的，最后也特判了一下是否为空区间，但实际上输入已经保证了区间不为空，所以直接将区间赋值为第一个区间左右端点即可。

---

[M区间问题] lc56. 合并区间(区间合并+贪心) 力扣上的本题原题。

成熟的模板：

#include <iostream>
#include <cstring>
#include <algorithm>

using namespace std;

typedef pair<int, int> PII;

int n;
int a, b;
vector<PII> seg, res;

int main() {
    cin >> n;
    for (int i = 0; i < n; i ++ ) cin >> a >> b, seg.push_back({a, b});
    
    sort(seg.begin(), seg.end());
    int l = seg[0].first, r = seg[0].second;
    for (int i = 0; i < seg.size(); i ++ ) {
        if (seg[i].first > r) res.push_back({l, r}), l = seg[i].first, r = seg[i].second;
        else r = max(r, seg[i].second);
    }
    res.push_back({l, r});
    
    cout << res.size() << endl;
    
    return 0;
}