【并查集】Graph practice T1 bricks 题解

Problem 1. bricks
Input file: bricks.in
Output file: bricks.out
Time limit: 1 second
jyb 在BUAA 天天被大神虐，所以只能去搬砖了。
终于在2019 年的夏天，菜菜的jyb 找不到工作，真的去工地搬砖了。jyb 的工头cky 是一个很麻烦的人，他
会让jyb 按某种方式搬砖，还问会问一些奇怪的问题。
现在有n 块砖，m 次操作。操作有两种：
1. M x y 把编号为x 的砖所在的一摞砖搬到编号为y 的砖所在的一摞砖的上面。如果x 和y 在同一摞砖则
忽略这个操作。（最初，每块砖都是单独一摞）
2. C x 询问x 下面压着多少块砖。
jyb 搬砖实在是太累了，想请你帮忙回答一下cky 工头的询问。
Input
第1 行，2 个整数n;m，表示一共有多少块砖以及有多少操作。
接下来m 行，每行一个操作，操作表示形式与前文一致。
Output
对于每次询问操作，输出答案
Sample
bricks.in bricks.out
6 6
M 1 6
C 1
M 2 4
M 2 6
C 3
C 4
1
0
2
Note
• 对于30% 的数据，1 n 103,1 m 103;
• 对于60% 的数据，1 n 104,1 m 105;
• 对于100% 的数据，1 n 2 105,1 m 3 105。

Bricks：并查集
用链表什么的暴力枚举题意，最坏复杂度是O(m^2)的。
注意到每次都是移动一摞砖，其实问题就是集合的合并，只是这个集合是一个有序的，我们还需要知道每个元素在集合中的位置。我们可以利用并查集高效的来解决。我们引入一个dep数组，对于祖先节点（将每摞砖最上方的一块设为祖先，设最下面的为祖先也是可以的），我们记录该集合中一共有多少块砖，对于其他节点，我们记录它上方有多少块砖，这样我们可以很容易地计算出下方有多少块。这样在合并x和y（设他们祖先为fx，fy）时，我们将dep[fx]赋给dep[fy]（因为移到了上方，fx有多少块fy上方就有多少块），然后dep[fx]要加上原本的dep[fy]。难点就是在合并的过程中，dep数组怎么维护。其实很简单，只需要在路径压缩时顺带维护即可，如果一个节点x的祖先fx不是祖先（即已被合并过），那么dep[x]加上dep[fx]即可（等于头上又多了一摞砖，所以要加上）。
时间复杂度O(α(m))

#include <iostream>
#include <cstdio>

using namespace std;

const int maxn = 2e5+5;
int n,m;
int fa[maxn],top[maxn],bri[maxn],num[maxn];

template <class T> inline void read(T &x) {
    int flag = 1; x = 0;
    char ch = getchar();
    while(ch <  '0' || ch >  '9') { if(ch == '-')  flag = -1; ch = getchar(); }
    while(ch >= '0' && ch <= '9') { x = (x<<1)+(x<<3)+ch-'0'; ch = getchar(); }
    x *= flag;
}

inline char Read() {
    char ch = getchar();
    while(ch == ' ' || ch == '\n') ch = getchar();
    return ch;
}

int root(int x) {       //返回最底下的砖 
    if(fa[x] != x) return root(fa[x]);
    else return x;
}

inline void unionn(int a, int b) {      //合并两堆砖 
    int ra = root(a), rb = root(b);
    for(int i = top[b]; ; i = fa[i]) {
        bri[i] += num[ra];
        if(i == rb) break;
    }
    fa[rb] = ra; top[a] = top[b]; num[ra] += num[rb];
}

int main() {
    freopen("bricks.in","r",stdin);
    freopen("bricks.out","w",stdout);
    read(n); read(m);
    for(register int i = 1; i <= n; i++) fa[i] = top[i] = i, num[i] = 1;
    while(m--) {
        char c = Read();
        if(c == 'M') {
            int a,b;
            read(a), read(b);
            if(root(a) != root(b)) unionn(b, a);
        } else {
            int b;
            read(b);
            printf("%d\n",bri[b]);
        }
    }
    return 0;
}