1329 - Corporative Network(并查集)

来源:互联网 发布:mysql indexof 编辑:程序博客网 时间:2024/05/18 15:29

题目链接

该题是比较经典的并查集题目,经典在需要在路径压缩时维护一些量。    由于每棵树除了根结点不能换之外,其他结点的位置可以任意改变,这恰好符合并查集的特点。

但是如果用最朴素的方法维护每个结点到根的距离,那么每次新加入的结点就要向上遍历一遍所有结点,时间复杂度难以承受(因为无法路径压缩) 。 所以我们可以在路径压缩的时候顺便求出d值 。   在每次返回某个结点的祖先之前先将当前结点的d值更新,即加上当前父结点的d值, 然后就可以路径压缩了。

所以这么做的原因是为了防止并查集退化成一条长长的链表,为了不牺牲路径压缩这个好东西。

细节参见代码:

#include<bits/stdc++.h>using namespace std;typedef long long ll;const int mod = 1000;const int maxn = 20000 + 5;const int INF = 1000000000;int T,n,m,x,y,p[maxn],d[maxn];void init(int v) {    for(int i=1;i<=v;i++) p[i] = i,d[i] = 0;}int findset(int x) {    if(p[x] == x) return x;    else {        int root = findset(p[x]);        d[x] += d[p[x]];        return p[x] = root;    }}char s[5];int main() {    scanf("%d",&T);    while(T--) {        scanf("%d",&n);        init(n);        while(~scanf("%s",s)&&s[0]!='O') {            if(s[0] == 'E') {                scanf("%d",&x);                findset(x);                printf("%d\n",d[x]);            }            else {                scanf("%d%d",&x,&y);                p[x] = y ; d[x] = abs(x - y)%mod;            }        }    }    return 0;}

0 0
原创粉丝点击