《大白书》192页 uva 1329 合作网络

【问题描述】

　　有n个结点（编号为1..n），初始时每个结点的父亲都不存在。你的任务是执行一次I操作和E操作，格式如下：

　　I u v：把节点u的父亲点设置为v，距离为|u-v|除以1000的余数。输入保证执行指令前u没有父亲节点。

　　E u：询问u 到根接点的距离（输出距离 mod 1000的结果）。

【输入格式】

　　输入第一行为测试数据组数T。每组数据第一行为n（5<=n<=20 000）。接下来有不超过20000行，每一行一条指令，以”O”结尾。I指令的个数不小于n。

【输出格式】

　　对于每条E指令，输出查询结果（输出距离 mod 1000的结果）。

【输入样例】

1

4

E 3

I 3 1

E 3

I 1 2

E 3

I 2 4

E 3

O

【输出样例】

0

2

3

5

【数据范围】

5≤n≤20000

【来源】

《大白书》192页 uva 1329

就是简单的并查集。自己看代码能懂的。

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<vector>
#include<algorithm>
#include<cstring>
using namespace std;
const int maxn=20005;
int n,q,fa[maxn],d[maxn];

void in()
{
for(int i=0;i<=n;i++)
fa[i]=i;
memset(d,0,sizeof(d));
}

int find(int x)
{
if(fa[x]==x)
return x;
int t=find(fa[x]);
d[x]=(d[x]+d[fa[x]])%1000;
fa[x]=t;
return t;
}

void work(int x,int y)
{
fa[x]=y;
d[x]=(abs(x-y))%1000;
}

int main()
{
//freopen("in.txt","r",stdin);
int T;
scanf("%d",&T);
for(int i=1;i<=T;i++)
{
scanf("%d",&n);
in();
char c[5];
scanf("%s",c);
while(1)
{
if(c[0]=='I')
{
int x,y;
scanf("%d%d",&x,&y);
work(x,y);
}
if(c[0]=='E')
{
int x;
scanf("%d",&x);
int ans=find(x);
printf("%d\n",d[x]);
}
if(c[0]=='O')
break;
scanf("%s",c);
}
}
return 0;
}