朋友（典型的并查集问题）

问题描述

在社交的过程中，通过朋友，也能认识新的朋友。
在某个朋友关系图中，假定 A 和 B 是朋友，B 和 C 是朋友，那么 A 和 C 也会成为朋友。即，我们规定朋友的朋友也是朋友。
现在，已知若干对朋友关系，询问某两个人是不是朋友。
请编写一个程序来解决这个问题吧。

输入描述

第一行：三个整数 n,m,p ((n≤5000,m≤5000,p≤5000)，分别表示有 n 个人，m 个朋友关系，询问 p 对朋友关系。
接下来 m 行：每行两个数 Ai ​ ,Bi ​，1≤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

思路

这就是一个典型的并查集问题

AC代码

#include<iostream>
using namespace std;
int father[5005],n;
void init(int n){
    for(int i = 1;i<=n;i++){
        father[i] = i;
    }
}
int get(int x){
    if(father[x]==x){
        return x;
    }
    return father[x] = get(father[x]);
}
void merge(int a,int b){
    a = get(a);
    b = get(b);
    if(a!=b){
        father[a] = b;
    }
}
int main(){

    int n,m,p;
    cin>>n>>m>>p;
    init(n);
    int a1,b1;
    int a2,b2;
    for(int i = 0;i<m;i++){
        cin>>a1>>b1;
        merge(a1,b1);
    }
    for(int k = 0;k<p;k++){
        cin>>a2>>b2;
        int x = get(a2);
        int y = get(b2);
        if(x==y){
            cout<<"Yes"<<endl;
        }else{
            cout<<"No"<<endl;
        }
    }
    return 0;
}