这个题大概意思就是描述子节点到根节点距离关系,但随着I的操作,每次根节点可能会改变,可以先把问题这么想,距离既然只是节点到根节点的关系,那么进行状态压缩是必要的,因为节点是如何到根节点的我们并不关心,但要是根节点也改变了,那么我们得在更新该节点到新根节点的距离前,把该节点的父亲节点到新根节点的距离给维护了,这就是为什么的先维护父亲节点,再来更新,否则会报错,一定要分清他们的先后顺序,否则会WA。
#include<stdio.h>
#include<stdlib.h>
#include<vector>
#define maxn 20005
int p[maxn];//代表父亲节点
int d[maxn];//代表到根节点距离
int find(int x){//并查集
if(p[x]!=x){//不是根节点
int root=find(p[x]);// 由下可知这一步是必不可少的,必须先把x点的父亲节点到新父亲节点距离更新才行,这里的先后顺序不能错
d[x]+=d[p[x]];//这里是更新d[x]的距离,在原距离基础上加上父亲节点到新的父亲节点距离
return p[x]=root;//返回
}
else return x;
}
int main(){
char cmd[5];
int n,a,b,t;
scanf("%d",&t);
while(t--){
scanf("%d",&n);
for(int i=0;i<=n;i++){ d[i]=0; p[i]=i; }//初始化
while(scanf("%s",cmd)==1&&cmd[0]!='O'){
if(cmd[0]=='E'){
scanf("%d",&a);
find(a);//状态压缩算距离
printf("%d\n",d[a]);
}
if(cmd[0]=='I'){
scanf("%d %d",&a,&b);
p[a]=b;
d[a]=abs(a-b)%1000;
}
}
}
return 0;
}