【JZOJ A组】 量子纠缠

Description

Input

第一行包含一个正整数 m，代表操作数。
接下来 m 行，每行可能有以下形式：
1 s 代表将数字串 s 加入信息集中
2 s 代表询问数字串 s 是否在信息集中
3 a b 代表使数字串 a 和 b 互相纠缠

Output

对于每一个 2 操作，如果询问串不在集合中，请输出一行一个整数 0，否则输出一行一个整 数 1。

Sample Input

11
1 123
2 123
2 0
3 12 13
1 124
2 133
2 134
2 13
3 1 11
2 111
2 11111111111111111111111124

Sample Output

1
0
1
1
0
0
1

Data Constraint

思路

对于前两个操作，一棵trie就可以解决。

对于纠缠操作，在trie上建并查集，做合并操作，详见代码

代码

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
const int maxn=8e6+77;
int tr[maxn][12],f[maxn],cnt,t;
char st[maxn*8];
bool mk[maxn];
int gf(int x)
{
    return x==f[x]?x:f[x]=gf(f[x]);
}
void insert(char *s)
{
    int l=strlen(s),u=1;
    for(int i=0; i<l; i++)
    {
        int v=s[i]-48;
        if(!tr[u][v]) tr[u][v]=++cnt;
        u=gf(tr[u][v]);
    }
    mk[u]=1;
}
bool query(char *s)
{
    int l=strlen(s),u=1;
    for(int i=0; i<l; i++)
    {
        int v=s[i]-48;
        if(!tr[u][v]) return 0;
        u=gf(tr[u][v]);
    }
    return mk[u];
}
void init()
{
    for(int i=1; i<=1000077; i++) f[i]=i;
    cnt++;
}
int &get(char *s)
{
    int u=1,l=strlen(s);
    for(int i=0; i<l-1; i++)
    {
        int v=s[i]-48;
        if(!tr[u][v]) tr[u][v]=++cnt;
        u=gf(tr[u][v]);
    }
    return tr[u][s[l-1]-48];
}
void merge(int &a,int &b,bool flag)
{
    a=gf(a);
    int y=gf(b);
    if(!a&&!y)
    {
        if(flag) a=b=++cnt; return;
    }else if(a==y) return;
    if(!a)
    {
        a=y; return;
    }else 
    if(!b)
    {
        b=a; return;
    }
    f[y]=a;
    if(mk[y]) mk[a]=1;
    for(int i=0; i<10; i++)
    {
        merge(tr[a][i],tr[y][i],0); a=gf(a);
    }
}
void work()
{
    int &u=get(st);
    scanf("%s",st);
    int &v=get(st);
    merge(u,v,1);
}
int main()
{
    freopen("quantum.in", "r", stdin);
    freopen("quantum.out", "w", stdout);
    init();
    scanf("%d",&t);
    while(t--)
    {
        int p;
//      printf("t=%d\n",t);
        scanf("%d%s",&p,st);
        if(p==1) insert(st);
        if(p==2) printf(query(st)?"1\n":"0\n");
        if(p==3) work();
    }
}