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

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

2014年03月09日 00:19:33

阅读数：914

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节点放在其所属连通分量的根节点下面。

#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;
}

#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;
}