关系网 （并查集）

题目描述

在一个社区里，每个人都有自己的小圈子，还可能同时属于很多不同的朋友圈。我们认为朋友的朋友都算在一个部落里，于是请你检查一下，对于给定的两个人是否属于同一个部落。

输入格式

第一行包含三个整数n, m, q (1≤n,m,q ≤105) 分别表示一共有 n 个人，m 次合并，q 次查询。

接下来的 m 行中的每一行包含两个整数 A 和 B 表示 A 和 B 是好朋友。

接下来 q 次询问，每行给出两个数字 A， B。

数据保证: A 和 B 合法。

输出格式

输出共 q 行。

对于每次询问，如果 A 和 B 属于同一部落输出"YES"，否则输出"NO"。

样例

输入数据 1

4 2 3
1 2
3 4
1 2
1 3
3 4

输出数据 1

YES
NO
YES

以前学区间合并的时候用区间合并写过，当时写了一大堆，结果也没过，现在就用并查集来写一下

代码： 

#include <iostream>
using namespace std;

const int N = 100010; 
int p[N]; //定义一个数组p，表示每个数字的父节点

//定义一个函数find，用于查找x所在的集合的根节点
int find(int x) {
    if (p[x] != x) p[x] = find(p[x]); //如果x不是根节点，就递归地查找它的父节点，并进行路径压缩
    return p[x]; //返回x的根节点
}

int main() {
    int n, m, q;
    cin >> n >> m >> q; 
    for (int i = 1; i <= n; i++) p[i] = i; //初始化每个数字的父节点为自己，即每个数字为一个单独的集合
    for (int i = 0; i < m; i++) 
    { 
        int a, b; 
        cin >> a >> b; 
        p[find(a)] = find(b); //将a所在的集合的根节点的父节点设为b所在的集合的根节点，即将两个集合合并
    }
    for (int i = 0; i < q; i++) {
        int a, b; 
        cin >> a >> b; 
        if (find(a) == find(b)) cout << "YES" << endl; //如果a和b所在的集合的根节点相同，说明它们在同一个区间内，输出YES
        else cout << "NO" << endl; 
    }
}