c++ 并查集模板

找朋友

在社交的过程中，通过朋友，也能认识新的朋友。在某个朋友关系图中，假定 A 和 B 是朋友，B 和 C 是朋友，那么 A 和 C 也会成为朋友。即，我们规定朋友的朋友也是朋友。

现在，已知若干对朋友关系，询问某两个人是不是朋友。

请编写一个程序来解决这个问题吧。

输入格式
第一行：三个整数 n,m,p(n≤5000,m≤5000,p≤5000)分别表示有n 个人，m 个朋友关系，询问p 对朋友关系。

接下来 m 行：每行两个数Ai,Bi1≤Ai,Bi≤N，表示Ai​ 和 Bi具有朋友关系。

接下来 p 行：每行两个数，询问两人是否为朋友。

输出格式
输出共 p 行，每行一个Yes或No。表示第i个询问的答案为是否朋友。

样例输入

6 5 3
1 2
1 5
3 4
5 2
1 3
1 4
2 3
5 6

样例输出
Yes
Yes
No

解答：

#include <iostream>

using namespace std;

int father[5001];
int get(int x){
    if (father[x]==x){
        return x;
    }
    return father[x]=get(father[x]);
}


int main(){
    int n,m,p;
    int a,b;
    cin>>n>>m>>p;
    for (int i=1;i<=n;i++){
        father[i]=i;
    }
    int temp=m;
    while (m--){
        cin>>a>>b;
        a=get(a);
        b=get(b);
        if (a!=b) {
            father[a] = b;
        }
    }

    for (int i=1;i<=p;i++){
        cin>>a>>b;
        a=get(a);
        b=get(b);
        if (a==b) {
            cout<<"YES"<<endl;
        }
        else{
            cout<<"NO"<<endl;
        }
    }

    return 0;
}