poj1988&&hdu2818 带权并查集

 

题意：

原来有n个东西，可以相互叠起来。有两种操作，M   a   b 将a所在的堆按原来的顺序叠到b所在的堆上面。C   a操作输出a下面有多少个东西。

 

解法：并查集

用down[i]记录在i的上面有多少个东西，用num[i]记录i所在的堆里共有多少个东西。则答案为down[i]

 

查询父节点压缩路径的时候更新down数组

 

#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
#include<queue>
#include<vector>
#include<stack>
#include<string>
#include<map>
#include<cmath>
using namespace std;

#define PB(x) push_back(x)
#define INF 0xfffffff
const int maxn=30005;

int f[maxn],down[maxn],num[maxn];
inline int get(int x){
    if(x==f[x]) return x;
    int t=f[x];
    f[x]=get(f[x]);
    down[x]+=down[t];
    return f[x];
}

inline void join(int a,int b){
    int fa=get(a);
    int fb=get(b);
    if(fa==fb) return;
    f[fa]=fb;
    down[fa]+=num[fb];
    num[fb]+=num[fa];
}

inline void init(){
    for(int i=0;i<maxn;++i){
        f[i]=i;
        num[i]=1;
        down[i]=0;
    }
}

int main()
{
    freopen("C:\\Users\\yk\\Desktop\\1.in","r",stdin);
    int p,a,b;
    char ch[3];
    while(~scanf("%d",&p)){
        init();
        while(p--){
            scanf("%s",ch);
            if(ch[0]=='M'){
                scanf("%d%d",&a,&b);
                join(a,b);
            }
            else {
                scanf("%d",&a);
                get(a);
            printf("%d\n",down[a]);
            }
        }
    }
    return 0;
}