关闭

指南 第三章 例题6 UVALive 3027 Corporative Network(并查集的应用)

标签: 哈理工OJ
160人阅读 评论(0) 收藏 举报
分类:

题目链接:http://7xjob4.com1.z0.glb.clouddn.com/8709392d143986024ba5de491366d376

下面写下中文题意:
有n个结点,初始时每个结点的父结点都不存在。你的任务是执行一次I操作和E操作,格式如下:
(1)I u v:把结点u的父结点设为v,距离为|u-v|除以1000的余数。输入保证执行指令前u没有父结点。
(2)E u:询问u到根结点的距离。
【输入格式】
输入第一行为测试数据组数T。每组数据第一行为n(5<=n<=20000)。接下来有不超过20000行,每行一条指令,以“O”结尾。I指令的个数小于n。
【输出格式】
对于每条E指令,输出查询结果。
【分析】
因为题目只查询结点到根结点的距离,所以每棵树除了根结点不能换之外,其他结点的位置可以任意改变,这恰好符合并查集的特点,但是需要记录附加信息。如果记录每个结点到根的距离,那么每次I操作都要更新很多结点的信息,时间复杂度难以保证,因此考虑记下每个结点到父结点的距离为d[i],然后在路径压缩时维护这个d数组。

下面是AC代码:

#include<cstdio>
#include<algorithm>
using namespace std;

const int maxn=20000 + 10;
int pre[maxn],d[maxn];

int findset(int x)//路径压缩,同时维护d[i];结点i到树根的距离
{
    if(x==pre[x])
    {
        return x;
    }
    else
    {
        int root=findset(pre[x]);
        d[x]+=d[pre[x]];
        return pre[x]=root;
    }
}

int main()
{
    int t;
    scanf("%d",&t);
    while(t--)
    {
        int n,u,v;
        char cmd[9];
        scanf("%d",&n);
        for(int i=1;i<=n;i++)//初始化,每个结点单独是一棵树
        {
            pre[i]=i;
            d[i]=0;
        }
        while(scanf("%s",cmd)&&cmd[0]!='O')
        {
            if(cmd[0]=='E')
            {
                scanf("%d",&u);
                findset(u);
                printf("%d\n",d[u]);
            }
            else
            {
                scanf("%d%d",&u,&v);
                pre[u]=v;
                d[u]=abs(u-v)%1000;
            }
        }
    }
    return 0;
}
0
0

查看评论
* 以上用户言论只代表其个人观点,不代表CSDN网站的观点或立场
    个人资料
    • 访问:153790次
    • 积分:7486
    • 等级:
    • 排名:第3089名
    • 原创:600篇
    • 转载:1篇
    • 译文:0篇
    • 评论:58条
    最新评论
    Java栏目
    JAVA学习与有关心得