UVALive 3027 Corporative Network

题意：有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;
}