算法 - 区间合并

最新推荐文章于 2023-12-01 23:55:21 发布

御用厨师

最新推荐文章于 2023-12-01 23:55:21 发布

阅读量109

点赞数

分类专栏：基础算法

本文链接：https://blog.csdn.net/qq_45520647/article/details/119543566

版权

数据结构 c++ 算法

基础算法专栏收录该内容

8 篇文章 1 订阅

订阅专栏

从左到右，每出现一个新区间进行区间合并时，有以下三种情况：（假设上一个区间是[st ~ ed]）

上一个：|----------|
这一个： |------|
这种情况不需要改变。
上一个：|----------|
这一个： |------------|
这种情况需要更新ed
上一个：|----------|
这一个： |--------|
这种情况需要更新st和ed，并记录上一个区间。

板子如下：

typedef pair<int, int> PII;
vector<PII> segs;

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

    sort(segs.begin(), segs.end()); //pair排序优先first

    int st = -2e9, ed = -2e9; //保证第一个区间可以处理即可 < -1e9均可
    for (auto seg : segs)
        if (ed < seg.first)
        {
        	// 情况3
        	// 避免最初[-2e9,-2e9]区间也合并进来
            if (st != -2e9) res.push_back({st, ed});
            
            st = seg.first, ed = seg.second;
        }
        //情况1和2
        else ed = max(ed, seg.second);
        
	//存最后一个区间
	//if是防止题目给空区间，但只有力扣会这么傻逼的给数据
    if (st != -2e9) res.push_back({st, ed});

    segs = res;
}

例题：区间合并
代码：

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

using namespace std;

typedef pair<int, int> PII;

vector<PII> segs;

void merge(vector<PII> &segs){
    vector<PII> res;
    sort(segs.begin(), segs.end());
    int st = -2e9, ed = -2e9;
    for(auto seg : segs){
        if(ed < seg.first){
            //情况3
            if(st != -2e9) res.push_back({st, ed}); //这里存的是上一个
            st = seg.first;
            ed = seg.second;
        }
        else
            ed = max(ed, seg.second);
    }
    
    //最后一个是没有存入的，或者是单独的，或者是和前面的合并的
    if(st != -2e9) res.push_back({st, ed});
    segs = res;
}

int main(){
    int n;
    cin >> n;
    int l, r;
    while(n --){
        scanf("%d%d", &l, &r);
        segs.push_back({l, r});
    }
    
    merge(segs);
    
    cout << segs.size();
    
    return 0;
}