题意:有n个结点、两个操作。第一个操作是将结点u的父结点设置为v(距离为|u-v|%1000),第二个操作是询问u到根结点的距离。
思路:并查集。用一个额外的数组储存当前结点到父结点的距离,不加路径优化2000多ms过了,速度慢了点,加优化的话就要维护数组了,应该速度会提升很多。
代码:
#include <iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<algorithm>
#include<map>
#include<queue>
#include<stack>
#include<cmath>
#include<vector>
#define inf 0x3f3f3f3f
#define Inf 0x3FFFFFFFFFFFFFFFLL
#define eps 1e-9
#define pi acos(-1.0)
using namespace std;
const int maxn=20000+10;
int parents[maxn],dis[maxn];
int Find(int x)
{
if(parents[x]==-1) return 0;
int sum=0;
sum=dis[x]+Find(parents[x]);
return sum;
}
void Uion(int u,int v)
{
parents[u]=v;
dis[u]=(abs(u-v))%1000;
}
int main()
{
//freopen("in.txt","r",stdin);
//freopen("out.txt","w",stdout);
int t,n;
char s[5];
cin>>t;
while(t--)
{
cin>>n;
for(int i=0;i<=n;++i)
{
parents[i]=-1;
dis[i]=0;
}
int a,b;
while(cin>>s)
{
if(s[0]=='O') break;
if(s[0]=='I')
{
cin>>a>>b;
Uion(a,b);
}
else
{
cin>>a;
cout<<Find(a)<<endl;
}
}
}
return 0;
}