bzoj3673 可持久化并查集by zky

3673: 可持久化并查集 by zky

Time Limit: 5 Sec   Memory Limit: 128 MB
Submit: 1471   Solved: 661
[ Submit][ Status][ Discuss]

Description

n个集合 m个操作
操作：
1 a b 合并a,b所在集合
2 k 回到第k次操作之后的状态(查询算作操作)
3 a b 询问a,b是否属于同一集合，是则输出1否则输出0

0<n,m<=2*10^4

Input

Output

Sample Input

5 6
1 1 2
3 1 2
2 0
3 1 2
2 1
3 1 2

Sample Output

1
0
1

HINT

Source

出题人大SB

可持久化线段树

这题目好鬼畜啊…可持久化并查集…

要维护可持久化并查集实际就是要维护可持久化数组，可持久化树组用可持久化线段树实现，将线段树的叶子节点看作数组的元素即可。

注意线段树叶子节点的每次修改都要新开节点。

#include<iostream>
#include<cstdlib>
#include<cmath>
#include<cstring>
#include<cstdio>
#include<algorithm>
#define F(i,j,n) for(int i=j;i<=n;i++)
#define D(i,j,n) for(int i=j;i>=n;i--)
#define ll long long
#define maxn 20005
#define maxm 3000000
using namespace std;
int n,m,p,x,y,cnt;
int rt[maxn],v[maxm],ls[maxm],rs[maxm];
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*10+ch-'0';ch=getchar();}
    return x*f;
}
void build(int &k,int l,int r)
{
    k=++cnt;
    if (l==r){v[k]=l;return;}
    int mid=(l+r)>>1;
    build(ls[k],l,mid);build(rs[k],mid+1,r);
}
int query(int k,int l,int r,int pos)
{
    if (l==r) return v[k];
    int mid=(l+r)>>1;
    if (pos<=mid) return query(ls[k],l,mid,pos);
    else return query(rs[k],mid+1,r,pos);
}
void insert(int x,int &y,int l,int r,int pos,int val)
{
    y=++cnt;
    if (l==r){v[y]=val;return;}
    int mid=(l+r)>>1;
    ls[y]=ls[x];rs[y]=rs[x];
    if (pos<=mid) insert(ls[x],ls[y],l,mid,pos,val);
    else insert(rs[x],rs[y],mid+1,r,pos,val);
}
int find(int &root,int x)
{
    int tmp=query(root,1,n,x); 
    if (tmp==x) return x;
    else
    {
        int ret=find(root,tmp);
        insert(root,root,1,n,x,ret);
        return ret;
    }
}
int main()
{
    n=read();m=read();
    build(rt[0],1,n);
    F(i,1,m)
    {
        p=read();
        if (p==1)
        {
            int x=read(),y=read(),fx=find(rt[i-1],x),fy=find(rt[i-1],y);
            if (fx==fy) rt[i]=rt[i-1];
            else insert(rt[i-1],rt[i],1,n,fx,fy);
        }
        else if (p==2)
        {
            int x=read();
            rt[i]=rt[x];
        }
        else
        {
            int x=read(),y=read(),fx=find(rt[i-1],x),fy=find(rt[i-1],y);
            if (fx==fy) puts("1");
            else puts("0");
            rt[i]=rt[i-1];
        }
    }
    return 0;
}