并查集路径压缩中,维护某节点到根节点的距离(依题意为abs(x-y)%1000)

即 d[i] +=d[father[i] ]

#include <iostream>
#include <algorithm>
#include <cmath>
using namespace std;
 const int N=1e5;
 int n,fa[N],d[N];
 
 int find(int x){
 	if(x==fa[x]) return x;
 	
 	int t=find(fa[x]);
 	d[x]+=d[fa[x]];
	fa[x]=t; 
	return t;
 }
 signed main(){
 	int cas,i,x,y;
 	char ch;
 	cin>>cas;
 	
 	while(cas--){
 		cin>>n;
 		for(i=1;i<=n;i++) fa[i]=i,d[i]=0;
 		while(cin>>ch&&ch!='O'){
 			if(ch=='I'){
 				cin>>x>>y; fa[x]=y; d[x]=abs(x-y)%1000;
			 }
			else{
				cin>>x; find(x); cout<<d[x]<<endl;
			}
		 }
	 }
 }
 
 
 

posted on 2022-10-14 13:34  towboat  阅读(16)  评论(0)    收藏  举报