uva1329 合作网络

【问题描述】

　　有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

在输入时可以不用字符串而直接输入单个字符再进行判断，这样空格和回车符都会输入而不对程序运行产生影响。

#include<cstdio>
#include<vector>
#include<cstring>
#include<iostream>
#include<algorithm>
#define maxn 100005
using namespace std;
int t,n,m,x,y,z;
bool ok;
int fa[maxn],dist[maxn];
struct data
{
    int a,b,l;
};
vector<data>g;

int task(int x)
{
    if(fa[x]==x) return x;
    int root=task(fa[x]);
    dist[x]=(dist[x]+dist[fa[x]])%1000;
    fa[x]=root;
    return root;
}
void clear()
{
    for(int i=0;i<=n;i++) fa[i]=i;
}
int main()
{
    //freopen("in.txt","r",stdin);
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d",&n);
        ok=false;
        char ch;
        clear();
        memset(dist,0,sizeof(dist));
        while(scanf("%c",&ch)==1)
        {
            if(ch=='E')
            {
                scanf("%d",&x);
                int ans=task(x);
                printf("%d\n",dist[x]);
            }
            if(ch=='I')
            {
                scanf("%d%d",&x,&y);
                fa[x]=y;
                dist[x]=abs(x-y)%1000;
            }
            if(ch=='O')
            {
                ok=true;
                break;
            }
            if(ok) break;
        }
    }
    return 0;
}