UVALive 3027---Corporative Network+并查集的应用

题目链接:点击进入
题意:
一开始有n个点,每个点都没有父节点。接下来不断执行I操作和E操作,直到执行O操作时结束;
I操作:I u v,u的父节点为v,路线长度为|u-v|%1000,保证u不存在起始父节点。
E操作:E u,查询u到其根节点的距离,输出。
题解:
并查集+路径压缩。
由于I操作时保证了u开始没有父节点,所以构成的是树;之后只是求点到跟的距离,所以只要用并查集即可,之后用路径压缩减少计算路径的时间。fa[x]一开始表示的是x的父节点,之后一次并查集查询后fa[x]表示的是x的根节点;在查询过程中d[x]表示x到fa[x]的距离,而在查找的时候还会维护所有路径上的点到根节点的距离,减少了下次查询的时间。

代码如下:

#include
#include
#include
#include
using namespace std;

#define maxn 22000
int fa[maxn],d[maxn];

int abs(int a)
{
     if(a<0)a=-a;
     return a;
}

int Find(int x)
{ ///注意这里的压缩顺序
    if(fa[x]!=x)
    {
        int rt=Find(fa[x]);
        d[x]+=d[fa[x]];
        return fa[x]=rt;
    }
    else
      return x;
}

int main()
{
      //freopen("in.txt","r",stdin);
      int t;
      scanf("%d",&t);
      while(t--)
      {
           int n;
           scanf("%d",&n);
           for(int i=1;i<=n;i++)
           {
                   fa[i]=i;
                   d[i]=0;
            }
           char str[10];
           while(scanf("%s",str))
           {
                int a,b;
                if(str[0]=='O') break;
                if(str[0]=='E')
                {
                    scanf("%d",&a);
                    Find(a);
                    printf("%d\n",d[a]);
                }
                else
                {
                    scanf("%d%d",&a,&b);
                    fa[a]=b;
                    d[a]=abs(a-b)%1000;
                }
           }
      }
  return 0;
}

你可能感兴趣的:(数据结构)