【算法模板】区间合并

目录

一、算法模板

二、情况介绍

三、例题

---

给定 n 个区间 [li,ri]，要求合并所有有交集的区间。

思路：可以将n个区间先按左端点排序，维护一个区间，并将维护区间与后面一个区间进行三种情况的比较（1.后一个区间包含于维护区间；2.后一个区间与维护区间部分相交；3.两个区间不相交），若后一个区间和当前维护的区间不相交时，需要将当前维护的区间存储到结果数组里，并将后一个区间作为新的维护区间，继续以上有关操作。

一、算法模板

// 将所有存在交集的区间合并
void merge(vector<PII> &segs)
{
    vector<PII> res;

    sort(segs.begin(), segs.end());// 按左端点排序

    int st = -2e9, ed = -2e9;// ed代表区间结尾，st代表区间开头
    for (auto seg : segs)
        if (ed < seg.first)// 情况1：两个区间无法合并
        {
            if (st != -2e9) res.push_back({st, ed});// 区间1放进res数组
            st = seg.first, ed = seg.second; // 维护区间2
        }
        else ed = max(ed, seg.second); // 情况2，3：能够合并，新区间右端点和远区间右端点谁大更新为谁   
        //注：排过序之后，不可能有区间2包含区间1

    if (st != -2e9) res.push_back({st, ed});// 放入最后一个区间，防止空数据
    

    segs = res;
}

二、情况介绍

• 先对所有区间按左端点排序，再看情况。
• 分以下三种情况：包含、相交、外面。

三、例题

AcWing 803.区间合并

给定 n 个区间 [li,ri]，要求合并所有有交集的区间。

注意如果在端点处相交，也算有交集。

输出合并完成后的区间个数。

例如：[1,3] 和 [2,6] 可以合并为一个区间 [1,6]。

输入格式

第一行包含整数 n。

接下来 n 行，每行包含两个整数 l 和 r。

输出格式

共一行，包含一个整数，表示合并区间完成后的区间个数。

数据范围

1≤n≤100000,

−109≤li≤ri≤109

输入样例：

5

1 2

2 4

5 6

7 8

7 9

输出样例：

3

代码如下:

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

using namespace std;

typedef pair<int, int> PII;

vector<PII> segs;

// 合并区间
void merge(vector<PII> &segs)
{
    // 按左端点排序
    sort(segs.begin(), segs.end());
    
    int st = -2e9, ed = -2e9;  // 初始化st和ed，st 为区间左端点，ed 为区间右端点
    vector<PII> res;   
    for (auto seg : segs) {
        if (ed < seg.first) {  // 情况1：不能合并
            if (st != -2e9) res.push_back({seg.first, seg.second});//把维护的区间放进res数组
            st = seg.first, ed = seg.second;  //更新维护的区间
        }
        else ed = max(ed, seg.second);  // 情况2，3：能够合并，新区间右端点和远区间右端点谁大更新为谁                                   // 不存在原区间在新区间中间的情况-> 以左端点排序保证
    }
    
    if (st != -2e9) res.push_back({st, ed});  //将最后一个维护区间放进res数组中
    
    segs = res;
}

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

对于区间，单个区间用pair<int,int> PII存，多个用vector存。
首先，要对所给区间按照开始位置由小到大进行排序。然后对于从前到后每个区间，将其开头与ed比较，如果比ed大并且此时ed！=-1e9，那么将{st，ed}加入到结果数组res中并更新st，ed；否则将ed进行更新。然后判断如果ed！=-1e9，那么将{st，ed}加入。最后输出res的长度即可。

---

 注意：这道题用贪心算法会更快。