2019牛客暑假多校训练赛第八场 E Explorer（线段树+并查集）

题目链接：https://ac.nowcoder.com/acm/contest/888/E

 

题意：给定n个点m条无向边，每条边有四个参数，两个端点编号，还有一个size的范围l,r。有多少合法的size可以从1走到n。

数据范围：1<=n,m<=1e5,1<=u<v<=n,1<=l<=r<=1e9

 

思路：首先对size离散化左端点正常离散化，右端点+1离散化，然后对于size建树，建树时线段树的每个节点代表一个左闭右开的区间。这里举例子说明一下，比如有size的范围是1-4（右端点+1是5）,4-9（右端点+1是10）,9-12（右端点+1是13），那么离散化以后就有6个点1,2,3,4,5,6,那么对于离散化后的1代表原来的1，2代表原来的4，3代表原来的5,4代表原来的9,5代表原来的10,6代表原来的13，那么开始建树虽然离散化完有6个节点但是我们只需要建5个叶子节点的树即可，因为树上的叶子节点1代表的是离散化后的[1,2)的左闭右开也就是原来的[1,4),叶子节点2代表[2,3)也就是原来的[4,5),叶子节点3代表[3,4)也就是原来的[5,9),叶子节点4代表[4,5)也就是原来的[9,10),叶子节点5代表[5,6)也就是原来的[10,13)因此用5个点完全的代表的[1,12]不重不漏。那么建树以后把对应的size的边插入到线段树里，在dfs搜一遍线段树，搜索时对于每一个节点搞一个按秩合并的并查集(因为涉及到把祖先还回去的操作所以不能压缩路径，比如路径中间有一个节点要把祖先还回去，那这个节点下面的节点都要把祖先还回去，然而这样并不能做到，所以只能按秩合并记录祖先)，若是1和n有公共的祖先就可以把线段树这个节点所代表的区间的贡献加入答案。

#include <bits/stdc++.h>
#define ls rt<<1
#define rs rt<<1|1
using namespace std;
typedef long long ll;
const int N=1e5+5;
vector<int>G[N<<3];
int n,m,ans,cnt;
int Rank[N],b[N<<1],l[N],r[N],u[N],v[N],f[N];
inline int fin(int x){return x==f[x]?x:fin(f[x]);}
inline void update(int rt,int L,int R,int l,int r,int x){
    if(l<=L&&R<=r){
        G[rt].push_back(x);
        return ;
    }
    int mid=(L+R)>>1;
    if(l<=mid)update(ls,L,mid,l,r,x);
    if(r>mid)update(rs,mid+1,R,l,r,x);
}
inline void dfs(int rt,int l,int r){
    vector<int>lastf;
    int len=G[rt].size(),mid=(l+r)>>1;
    for(int i=0;i<len;i++){
        int x=u[G[rt][i]],y=v[G[rt][i]];
        int fx=fin(x),fy=fin(y);
        ///按秩合并，不能压缩路径，这样才能在dfs完这个节点把已经连好的祖先撤销掉
        if(Rank[fx]>Rank[fy])lastf.push_back(fy),f[fy]=fx;
        else{
            lastf.push_back(fx),f[fx]=fy;
            if(Rank[fx]==Rank[fy])Rank[fy]++;
        }
    }
    if(fin(1)==fin(n))ans+=b[r+1]-b[l];
    else if(l<r)dfs(ls,l,mid),dfs(rs,mid+1,r);
    len=lastf.size();
    for(int i=len-1;i>=0;i--)f[lastf[i]]=lastf[i];
    ///撤销祖先
    ///父亲的深度不用还，太难还了，不还复杂度影响也不会很大
    lastf.clear();
}
int main(){
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++)f[i]=i,Rank[i]=1;
    for(int i=1;i<=m;i++){
        scanf("%d%d%d%d",&u[i],&v[i],&l[i],&r[i]);
        b[++cnt]=l[i],b[++cnt]=r[i]+1;
    }
    ///对离散化后的区间段左闭右开建树，每个节点代表一个左闭右开的区间，不重不漏
    sort(b+1,b+1+cnt);cnt=unique(b+1,b+1+cnt)-b-1;
    for(int i=1;i<=m;i++)
    update(1,1,cnt-1,lower_bound(b+1,b+1+cnt,l[i])-b,lower_bound(b+1,b+1+cnt,r[i]+1)-b-1,i);
    dfs(1,1,cnt-1);printf("%d\n",ans);
    return 0;
}