题目链接:http://acm.hdu.edu.cn/showproblem.php?pid=3172点击打开链接
题目描述:
输入N个人的关系,输出你最新输入的两个人所在社交分组的人数(相互有朋友关系的视为一个组)。
解题思路:
另设一个RANK数组来存储对应集合的元素个数,在merge合并操作的同时将数组合并即得到答案。对字符串处理用MAP,不然会超时,我就T了N多遍,后来干脆去现学现卖了MAP。
AC代码:
#include <cstdio>
#include <iostream>
#include <cstring>
#include <map>
using namespace std;
map <string,int>my_map;
int arr[100050],mem[100050];
int q;
void reset()
{
my_map.clear();
q=1;
memset(arr,0,sizeof(arr));
int i;
for(i=0;i<100050;i++)
{
arr[i]=i;
mem[i]=1;
}
}
int find(int x)
{
return x==arr[x]?x:arr[x]=find(arr[x]);
}
void mer(int x,int y)
{
int fx=find(x);
int fy=find(y);
if(fx!=fy&&mem[fx]<=mem[fy])
{
mem[fy]=mem[fx]+mem[fy];
arr[fx]=fy;
}
else if(fx!=fy&&mem[fx]>mem[fy])
{
mem[fx]=mem[fx]+mem[fy];
arr[fy]=fx;
}
}
int main()
{
reset();
int T,x,y,i;
char a[22],b[22];
while(scanf("%d",&T)!=EOF)
while(T--)
{
int n;
scanf("%d",&n);
while(n--)
{
scanf("%s%s",a,b);
if(!my_map[a])my_map[a]=q++;
if(!my_map[b])my_map[b]=q++;
x=my_map[a];
y=my_map[b];
mer(x,y);
printf("%d\n",mem[find(x)]);
}
reset();
}
return 0;
}
AC截图: