HDU 3172 并查集+trie树

点击打开链接

题意：给n个关系，然后问u与v的集合中元素的个数

思路：就是简单的并查集就行了，只是这个给的字符串比较猥琐，我们用trie树查找就行了，操作还是很简单的，然后注意的是出题人竟然设置了多组，还要多组输入，只能感叹，WA了两次，org~~~

#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <iostream>
#include <algorithm>
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
const int inf=0x3f3f3f3f;
const ll INF=0x3f3f3f3f3f3f3f3fll;
const int maxn=400010;
int f[maxn],num[maxn];
int find1(int x){
    if(x==f[x]) return f[x];
    int t=f[x];
    f[x]=find1(f[x]);
    return f[x];
}
void unite(int a,int b){
    int aa=find1(a);
    int bb=find1(b);
    if(aa==bb) return ;
    f[aa]=bb;
    num[bb]+=num[aa];
}
struct Trie{
    Trie *next[70];
    int v;
};
Trie *root;
char str1[30],str2[30];
void creatTrie(char *str,int idx){
    int len=strlen(str);
    Trie *p=root,*q;
    for(int i=0;i<len;i++){
        int id=str[i]-'A';
        if(p->next[id]==NULL){
            q=(Trie *)malloc(sizeof(Trie));
            q->v=0;
            for(int j=0;j<70;j++)
                q->next[j]=NULL;
            p->next[id]=q;
        }
        p=p->next[id];
    }
    p->v=idx;
}
int findtrie(char *str){
    int len=strlen(str);
    Trie *p=root;
    for(int i=0;i<len;i++){
        int id=str[i]-'A';
        p=p->next[id];
        if(p==NULL) return 0;
    }
    return p->v;
}
int dealtrie(Trie *T){
    if(T==NULL) return 0;
    for(int i=0;i<70;i++) if(T->next[i]!=NULL) dealtrie(T->next[i]);
    free(T);
    return 0;
}
int main(){
    int T,n;
    while(scanf("%d",&T)!=-1){
    while(T--){
        scanf("%d",&n);
        for(int i=0;i<=2*n;i++){
            f[i]=i;num[i]=1;
        }
        int k=1;
        root=(Trie *)malloc(sizeof(Trie));
        for(int i=0;i<70;i++) root->next[i]=NULL;
        for(int i=0;i<n;i++){
            scanf("%s%s",str1,str2);
            int ans1=findtrie(str1);
            int ans2=findtrie(str2);
            if(ans1==0) creatTrie(str1,k++);
            if(ans2==0) creatTrie(str2,k++);
            int id1=findtrie(str1);
            int id2=findtrie(str2);
            unite(id1,id2);
            int ans=find1(id1);
            printf("%d\n",num[ans]);
        }
        dealtrie(root);
    }
    }
    return 0;
}