UVA 1329 Corporative Network(并查集：路径压缩)

UVA 1329 Corporative Network(并查集：路径压缩)
题意：
初始时有n个点(编号1到n)，每个点都没有父节点，然后依次执行下面两条命令：
I u v ：把节点u的父亲设置为v，并且设u节点到v的距离为|u-v|%1000，输入保证执行I命令的时候，u没有父节点。
E u：查询u节点到根节点的距离。
输入：首先是T，表示实例个数。每个实例第一行是n（5<=n<=20000），接下有不超过200000行，每行一个指令，以单独的O字母行表示结束。
输出：对于每条E指令，输出查询结果。
分析：
本题连通分量的节点不仅要维护fa信息，还需要维护dist信息。
其中fa[i]表示i节点的直属父亲节点编号，dist[i]表示i节点到它直属父亲节点的距离。在每次findset(i)操作后，将i节点连接到当前连通分量的根节点下面，并且更新dist[i]的值。其实就是比普通的并查集操作多了一个相对距离的概念。
注意：每次查询u节点到根节点距离前，需要findset(u)操作，这样才能让u节点放在其所属连通分量的根节点下面。
AC代码(新)：

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int maxn=20000+5;

//路径压缩并查集
int fa[maxn];
int dist[maxn];
int findset(int x)
{
    if(fa[x]==-1) return x;

    int root = findset(fa[x]);
    dist[x] += dist[fa[x]];
    return fa[x] = root;
}

int main()
{
    int T;
    scanf("%d",&T);
    while(T--)
    {
        int n;
        scanf("%d",&n);
        for(int i=1;i<=n;i++)
        {
            fa[i]=-1;
            dist[i]=0;
        }

        char str[100];
        while(scanf("%s",str)==1 && str[0]!='O')
        {
            if(str[0]=='I')
            {
                int u,v;
                scanf("%d%d",&u,&v);
                fa[u]=v;
                dist[u]= abs(u-v)%1000;
            }
            else if(str[0]=='E')
            {
                int u;
                scanf("%d",&u);
                findset(u);
                printf("%d\n",dist[u]);
            }
        }
    }
    return 0;
}

AC代码：

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int MAXN=20000+100;
int d[MAXN];//点到根节点的距离
int pa[MAXN];//父节点
int findset(int x)//查询x的根节点，并且使得d[x]变成x到根的距离
{
    if(x==pa[x])return x;
    int root = findset(pa[x]);//执行完这句，pa[X]到根的距离已经是d[pa[x]]了
    d[x]=d[x]+d[pa[x]];
    return pa[x]=root;
}

int main()
{
    int T;
    scanf("%d",&T);
    while(T--)
    {
        int n;
        scanf("%d",&n);
        for(int i=0;i<=n;i++)
        {
            pa[i]=i;
            d[i]=0;
        }
        char op[20];
        while(scanf("%s",op))
        {
            if(op[0]=='O')
                break;
            else if(op[0]=='I')
            {
                int u,v;
                scanf("%d%d",&u,&v);

                pa[u]=v;
                d[u]=abs(u-v)%1000;
            }
            else if(op[0]=='E')
            {
                int u;
                scanf("%d",&u);
                findset(u);
                printf("%d\n",d[u]);
            }
            getchar();//读/n
        }
    }
    return 0;
}

new AC code：

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
using namespace std;
const int maxn=20000+5;

//并查集操作
int fa[maxn];
int dist[maxn];//dist[i]==x表i到其父节点的距离为x
int findset(int x)
{
    if(fa[x]==-1) return x;
    int father=fa[x];
    fa[x]=findset(fa[x]);//将x连接到连通分量根下面
    dist[x]=dist[x]+dist[father];//更新x到新父亲的距离
    return fa[x];
}

int main()
{
    int T;
    scanf("%d",&T);
    while(T--)
    {
        int n;
        scanf("%d",&n);
        memset(fa,-1,sizeof(fa));
        memset(dist,0,sizeof(dist));

        char cmd;
        int u,v;
        while(scanf(" %c",&cmd)==1)
        {
            if(cmd=='O') break;
            else if(cmd=='E')
            {
                scanf("%d",&u);
                findset(u);//这步不能少
                printf("%d\n",dist[u]);
            }
            else if(cmd=='I')
            {
                scanf("%d%d",&u,&v);
                fa[u]=v;
                dist[u]=abs(u-v)%1000;
            }
        }
    }
    return 0;
}