【分治 + 并查集】2019牛客暑期多校训练营（第八场）- E - Explorer

最新推荐文章于 2019-08-22 15:48:42 发布

原创最新推荐文章于 2019-08-22 15:48:42 发布 · 217 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

数据结构专栏收录该内容

33 篇文章

订阅专栏

本文探讨了一种结合图论与并查集的数据结构应用，通过离散化和线段树优化，解决特定路径寻找问题。利用带权并查集维护路径状态，实现高效的回溯操作。适用于解决具有区间限制的图遍历问题。

题目链接https://ac.nowcoder.com/acm/contest/888#question

题意

给出一张图，每条边都有一个区间。只有在区间内的数才能经过这条边。问有多少个数使得能从点1走到点n

题解

离散之后，在线段树上添加每条边。然后在线段树上dfs分治，如果当前区间添加的边（包括自己区间的边和所有祖先上的边）已经使得1和n在一个并查集内，则直接添加答案return。否则继续看子区间。
用带权并查集维护，因为回溯的时候需要把边拆掉。

#include<bits/stdc++.h>
#define sec second
#define fir first
using namespace std;
typedef long long ll;
typedef pair<int,int>piir;
const ll mod=1e9+7;
const ll N=2e5+7;
struct Edge{
    int u,v,l,r;
}e[N];
vector<int>vc[N<<2];
int xx[N*2],tot;
int ans;
int fa[N],sz[N];
int lst[N];
int n,m;
int fd(int x){return x==fa[x]?x:fd(fa[x]);}
int uni(int x,int y){
    int fx=fd(x),fy=fd(y);
    if(fx==fy) return 0;
    if(sz[fx]<sz[fy]){
        fa[fx]=fy;
        sz[fy]+=sz[fx];
        return fx;
    }
    else{
        fa[fy]=fx;
        sz[fx]+=sz[fy];
        return fy;
    }
}
void upd(int rt,int l,int r,int L,int R,int x){
    if(L<=l&&r<=R){
        vc[rt].push_back(x);
        return;
    }
    int m=l+r>>1;
    if(L<m) upd(rt<<1,l,m,L,R,x);
    if(m<R) upd(rt<<1|1,m,r,L,R,x);
}
void dfs(int rt,int l,int r){
    int sv=vc[rt].size();
    for(int i=0;i<sv;i++){
        int p=vc[rt][i];
        lst[p]=uni(e[p].u,e[p].v);
    }
    if(fd(1)==fd(n)) ans+=xx[r]-xx[l];
    else if(l+1<r){
        int m=l+r>>1;
        dfs(rt<<1,l,m);
        dfs(rt<<1|1,m,r);
    }
    for(int i=0;i<sv;i++){
        int p=vc[rt][i];
        int u=lst[p];
        int v=fa[u];
        fa[u]=u;
        sz[v]-=sz[u];
    }
}
int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<=m;i++){
        scanf("%d%d%d%d",&e[i].u,&e[i].v,&e[i].l,&e[i].r);e[i].r++;
        xx[++tot]=e[i].l;xx[++tot]=e[i].r;
    }
    sort(xx+1,xx+1+tot);
    tot=unique(xx+1,xx+1+tot)-xx-1;
    for(int i=1;i<=m;i++){
        e[i].l=lower_bound(xx+1,xx+1+tot,e[i].l)-xx;
        e[i].r=lower_bound(xx+1,xx+1+tot,e[i].r)-xx;
        upd(1,1,tot,e[i].l,e[i].r,i);
    }
    for(int i=1;i<=n;i++) fa[i]=i,sz[i]=1;
    dfs(1,1,tot);
    printf("%d\n",ans);
}