来自《算法竞赛入门经典训练指南》
1.题目原文
有n个节点,初始时每个节点的父亲节点都不存在。你的任务是执行一次I操作和E操作,格式如下:
I u v:把节点u的父亲节点设置为v,距离为|u-v|除以1000的余数.输入保证执行指令前u没有父亲节点
E u:询问u到父亲节点的距离。
2.解题思路
题目显然符合并查集的特点,但是和基本并查集不同的是需要查询以及维护每个节点到根节点的距离。在模板的基础上改一下就可以了。
3.AC代码
#include <algorithm>
#include <cctype>
#include <cmath>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <iomanip>
#include <iostream>
#include <map>
#include <queue>
#include <string>
#include <set>
#include <vector>
#include<cmath>
#include<bitset>
#include<sstream>
using namespace std;
#define INF 0x7fffffff
#define maxn 20005
//并查集
int parent[maxn];//父亲节点
int d[maxn];//到根节点的距离
//函数返回父亲节点
//路径压缩,同时维护数组d
int findset(int x)
{
if(parent[x]!=x){
int root=findset(parent[x]);
d[x]+=d[parent[x]];
return parent[x]=root;
}
else return x;
}
int main()
{
int t;
scanf("%d",&t);
while(t--){
int n,u,v;
scanf("%d",&n);
for(int i=0;i<n;i++){
parent[i]=i;
d[i]=0;
}
char c;
while(cin>>c&&c!='O'){
if(c=='I'){
scanf("%d%d",&u,&v);
parent[u]=v;
d[u]=abs(u-v)%1000;
}
else if(c=='E'){
scanf("%d",&u);
findset(u);
printf("%d\n",d[u]);
}
}
}
return 0;
}