关闭

UVa 1329 Corporative Network(并查集)

70人阅读 评论(0) 收藏 举报
分类:

题目链接:http://acm.hust.edu.cn/vjudge/problem/36132

大意:给出n个结点,一系列操作。对于操作I,输入u,v,把u的父结点设为v,距离为|u-v|mod1000;对于操作E,输入u,输出u到根结点的距离。

思路:两个操作正好对应了并查集的合并和查找,对于操作I,由于v没有父结点,直接令pre[u] = v即可;对于操作E,由于根节点是动态变化的,所以要先更新再查询。


#include<cstdio>
#include<cstring>
#include<iostream>
#include<cmath>
#include<algorithm>
using namespace std;
const int maxn = 2e4 + 5;
int n, pre[maxn], dist[maxn];

int find(int x)
{
    if (pre[x] == x) return pre[x];
    int y = pre[x];
    pre[x] = find(pre[x]);
    dist[x] += dist[y];
    return pre[x];
}

int main()
{
    int T;
    scanf("%d",&T);
    while(T--) {
        scanf("%d",&n);
        memset(dist, 0, sizeof dist);
        for(int i = 1; i <= n; i++) pre[i] = i;
        char s[2];
        while(scanf("%s",s) && s[0] != 'O') {
            if (s[0] == 'E') {
                int x;
                scanf("%d",&x);
                find(x);
                printf("%d\n",dist[x]);
            }
            else {
                int a,b;
                scanf("%d%d",&a,&b);
                pre[a] = b;
                dist[a] = abs(a-b) % 1000;
            }
        }
    }
}


0
0

查看评论
* 以上用户言论只代表其个人观点,不代表CSDN网站的观点或立场
    个人资料
    • 访问:9266次
    • 积分:1021
    • 等级:
    • 排名:千里之外
    • 原创:97篇
    • 转载:4篇
    • 译文:0篇
    • 评论:0条
    文章分类