【并查集的合并】AtCoder - 2159 連結 / Connectivity

Problem Description

给你n个城市，城市之间有道路，有铁路。
给你K条道路，L条铁路。
问你 对于每个点，有多少个点既可以只通过道路到达它，又可以只通过铁路到达它，自己达到自己直接算上。
例：
4 3 1
1 2
2 3
3 4
2 3
1点 自己到达自己
2点 自己到达自己，3点可以通过两种方式到达它
3点 自己到达自己，2点可以通过两种方式到达它
4点 自己到达自己
所以输出：
1 2 2 1

思路：

第一反应的思路就是，把道路能够相互到达的点，用并查集并起来。铁路能够相互到达的点，用并查集并起来。然后对于一个点，判断它们铁路的并查集 和 道路的并查集有多少个相同的点，相同点既是这个点的结果。

#include<bits/stdc++.h>
using namespace std;
const int N = 2*1e5;
int n;
int Find(int *f, int x)
{
    if(f[x] == x)
        return x;
    else
    {
        f[x] = Find(f, f[x]);
        return f[x];
    }
}
void Merge(int *f, int x, int y)
{
    x = Find(f, x);
    y = Find(f, y);
    if(x != y)
        f[y] = x;
}
int f[N], g[N], vis[N];
vector<int> vec[N];
map<int, int> q;
void init()
{
    for(int i = 0; i <= n; i++)
        f[i] = g[i] = i;
}
int main()
{
    int k, l, u, v;
    scanf("%d %d %d", &n, &k, &l);
    init();
    while(k--)
    {
        scanf("%d %d", &u, &v);
        Merge(f, u, v);//道路
    }
    while(l--)
    {
        scanf("%d %d", &u, &v);
        Merge(g, u, v);//铁路
    }
    memset(vis, 0, sizeof(vis));
    for(int i = 1; i <= n; i++) vec[Find(f, i)].push_back(i);//道路同一个并集合都指向同一个根
    for(int i = 1; i <= n; i++)
    {
        q.clear();
        for(int j = 0; j < vec[i].size(); j++)//枚举道路的其中一个并集合的所有点
        {
            q[Find(g, vec[i][j])]++;//让铁路存在该点的并集合 ++
        }
        for(int j = 0; j < vec[i].size(); j++)
        {
            vis[vec[i][j]] += q[Find(g, vec[i][j])];//集合里面有多少个重复的点，就是该点的结果。
        }
    }
    for(int i = 1; i <= n; i++)
    {
        printf("%d", vis[i]);
        if(i == n) printf("\n");
        else printf(" ");
    }
    return 0;
}