【模板】可持久化并查集

最新推荐文章于 2022-12-07 02:00:00 发布

weixin_34304013

最新推荐文章于 2022-12-07 02:00:00 发布

阅读量100

点赞数

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/PaperCloud/p/10161239.html

传送门

Solution

路径压缩的话，修改过于频繁，反而TLE掉
只要按秩合并就行了
其实就是可持久化数组？维护fa[i]和dep[i]

Code

bzoj 3673 可持久化并查集 by zky

#include<bits/stdc++.h>
#define ll long long
#define max(a,b) ((a)>(b)?(a):(b))
#define min(a,b) ((a)<(b)?(a):(b))
inline int read()
{
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){x=(x<<3)+(x<<1)+ch-'0';ch=getchar();}
    return x*f;
}
#define MN 100005
#define MM 3000005 
int n,m,fa[MM],dep[MM],rt[MN];
int sz,ls[MM],rs[MM];
#define mid ((l+r)>>1)
void Build(int &x,int l,int r)
{
    x=++sz;if(l==r)return (void)(fa[x]=l);
    Build(ls[x],l,mid);Build(rs[x],mid+1,r);
}
void Merge(int &x,int ori,int l,int r,int p,int f)
{
    x=++sz;ls[x]=ls[ori];rs[x]=rs[ori];
    if(l==r) return (void)(fa[x]=f,dep[x]=dep[ori]);
    p<=mid?Merge(ls[x],ls[ori],l,mid,p,f):Merge(rs[x],rs[ori],mid+1,r,p,f);
}
void update(int x,int l,int r,int p)
{
    if(l==r) return (void)(dep[x]++);
    p<=mid?update(ls[x],l,mid,p):update(rs[x],mid+1,r,p); 
}
int Query(int x,int l,int r,int p)
{
    if(l==r) return x;
    return p<=mid?Query(ls[x],l,mid,p):Query(rs[x],mid+1,r,p);
}
int getf(int x,int p)
{
    register int tmp=Query(x,1,n,p);
    return fa[tmp]==p?tmp:getf(x,fa[tmp]);
}
int main()
{
    register int i,opt,x,y;
    n=read();m=read();
    Build(rt[0],1,n);
    for(i=1;i<=m;++i)
    {
        opt=read();x=read();
        if(opt==1)
        {
            y=read();rt[i]=rt[i-1];
            x=getf(rt[i],x);y=getf(rt[i],y);
            if(fa[x]!=fa[y])
            {
                if(dep[x]>dep[y]) std::swap(x,y);
                Merge(rt[i],rt[i-1],1,n,fa[x],fa[y]);
                if(dep[x]==dep[y]) update(rt[i],1,n,fa[y]);
            }
        }
        else if(opt==2) rt[i]=rt[x];
        else if(opt==3)
        {
            y=read();rt[i]=rt[i-1];
            x=getf(rt[i],x),y=getf(rt[i],y);
            puts(fa[x]==fa[y]?"1":"0");
        }
    }
    return 0;
}

Blog来自PaperCloud，未经允许，请勿转载，TKS！

转载于:https://www.cnblogs.com/PaperCloud/p/10161239.html