解题思路:典型的并查集,只是每个人的名字要转换成数字,可以用map,也可以用字典树,我最开始用的字典树结果爆内存了。。
爆内存:
#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
const int maxn = 200000;
int n,fa[maxn],trie[maxn][52],cnt,id[2000000],num;
int tot[maxn];
char str[2][20];
void init()
{
for(int i = 0; i <= 2*n; i++)
{
fa[i] = i;
tot[i] = 1;
}
memset(id,-1,sizeof(id));
memset(trie,-1,sizeof(trie));
cnt = num = 0;
}
int insert(char *s)
{
int p = 0,idx;
while(*s != 0)
{
if(*s >= 'A' && *s <= 'Z') idx = *s - 'A' + 26;
else idx = *s - 'a';
if(trie[p][idx] == -1)
trie[p][idx] = ++cnt;
p = trie[p][idx];
s++;
}
if(id[p] == -1)
id[p] = ++num;
return id[p];
}
int find(int x)
{
if(fa[x] == x) return x;
return fa[x] = find(fa[x]);
}
int main()
{
int t,x,y,fx,fy;
scanf("%d",&t);
while(t--)
{
scanf("%d",&n);
init();
for(int i = 1; i <= n; i++)
{
scanf("%s %s",str[0],str[1]);
x = insert(str[0]);
y = insert(str[1]);
fx = find(x);
fy = find(y);
if(fx != fy)
{
fa[fy] = fx;
tot[fx] += tot[fy];
}
printf("%d\n",tot[fx]);
}
}
return 0;
}
别人的:
#include<stdio.h>
#include<map>
#include<iostream>
using namespace std;
map<string,int> m;
int set[100005];
int num[100005];
int find(int x)
{
int r=x;
while(r!=set[r])
r=set[r];
int i=x;
while(i!=r)
{
int j=set[i];
set[i]=r;
i=j;
}
return r;
}
void merge(int x,int y)
{
int fx=find(x);
int fy=find(y);
if(fx!=fy)
{
set[fx]=fy;
num[fy]+=num[fx];
printf("%d\n",num[fy]);
}
else
{
printf("%d\n",num[fy]);
}
}
int main()
{
int t;
char a[25];
char b[25];
while(scanf("%d",&t)!=EOF)
{
while(t--)
{
int n;
scanf("%d",&n);
for(int i=1;i<100005;i++)
{
set[i]=i;
num[i]=1;
}
m.clear();
int ans=1;
for(int i=1;i<=n;i++)
{
scanf("%s%s",a,b);
if(!m[a])
{
m[a]=ans++;
}
if(!m[b])
{
m[b]=ans++;
}
merge(m[a],m[b]);
}
}
}
return 0;
}