UVA 1329 - Corporative Network

求节点到根结点的距离。用一个D数组保存每一个节点到根结点的距离。

在I操作的时候路径压缩，更新D数组。

#include<stdio.h>#include<string.h>int abs(int x){return x>0?x:-x;}const int maxn=20000+20;int pa[maxn],d[maxn];int find(int x){if(pa[x]!=x){int root=find(pa[x]);d[x]+=d[pa[x]];//更新这一点到达根节点的距离。//printf("YY %d",x);return pa[x]=root;//这一步是更新父节点}else return x;}int main(){int tt;scanf("%d",&tt);while(tt--){for(int i=0;i<=maxn;i++){pa[i]=i,d[i]=0;}int n;scanf("%d",&n);char ch[10];while(scanf("%s",ch)!=EOF){if(ch[0]=='O') break;if(ch[0]=='E'){int x;scanf("%d",&x);find(x);printf("%d\n",d[x]);}else if(ch[0]=='I'){int x,y;scanf("%d%d",&x,&y);pa[x]=y;d[x]=abs(x-y)%1000;}}}return 0;}