题意:
两个操作:I u v:把节点u的父节点设为v,距离为|u-v|%1000。输入的u无父节点。
E u:询问u到根节点的距离。
题解:
这个可以说是带权并查集的母题了,只不过这里的距离是最直观的节点到根节点的距离,开个权值数组,记录每个节点到根节点的距离即可。
像食物链那类的带权并查集题,距离就是显得抽象了点,思想都在这道题里了。
#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<algorithm>
#include<queue>
#include<string>
#include<cstring>
#include<vector>
#include<functional>
#include<utility>
#include<set>
#include<map>
#include<cmath>
#include<stack>
using namespace std;
const int maxn=20000+10;
int pa[maxn],d[maxn];
int findset(int x)
{
if(pa[x]==-1)
return x;
else
{
int root=findset(pa[x]);
d[x]+=d[pa[x]];//更新高度
return pa[x]=root;
}
}
int main()
{
int T;
cin>>T;
while(T--)
{
int n,u,v;
char s[15];
cin>>n;
for(int i=1;i<=n;i++){pa[i]=-1;d[i]=0;}
while(scanf("%s",s)&&s[0]!='O')
{
if(s[0]=='E'){cin>>u;findset(u);cout<<d[u]<<endl;}
if(s[0]=='I') {cin>>u>>v;pa[u]=v;d[u]=abs(u-v)%1000;}
}
}
}