并查集应用

题目链接

并查集介绍

---

并查集是一种数据结构，用来管理元素分组情况，在对元素进行合并操作或查询两个元素是否属于同一组时十分高效。但无法对元素集合进行分割操作。

题目大意

---

给出n个人，编号分别是1~n，然后给m组关系，判断m组关系是否合理。从这样的描述中可以看处两种操作，把两个元素合并为一个集合（本题中，互为朋友的两个人在一个集合内， 可以比喻成一个朋友圈）、查询任意两个元素之间是否存在关系（本题中是朋友关系）。字里行间透露出一个字：并查集。

样例

---

这道题的题目意思不是非常清楚。
可以通过对样例的分析来理解。

样例一

4 3
1 3
3 4
1 4
给出了三种关系分别是：1-3，3-4，1-4。
意思使1和3是朋友，3和4是朋友，1和4是朋友。
用图表示是这样的

有两个集合（或者可以比喻成朋友圈），没有给出2号元素与其他元素的关系，所以2号元素是一个孤立的元素，也是一个只有自己的集合， 可见2号是非常孤独的。所以总的关系数目为3.

样例二

4 4
3 1
2 3
3 4
1 2
如果用图表示出样例给出的关系：

根据题目说的合理情况，4号元素应该和1号元素和2号元素都是朋友。
即正确的关系是这样的：

而题目的输入没有给出1-4，2-4这样的关系，所以这种情况是不合理的。
由此也可以看出一个集合内的任意两个元素都是有关系的。

样例三

10 4
4 3
5 10
8 9
1 2
如果画出关系图：

可以认为，有4个集合，每个集合有两个元素互为朋友关系。所以总的关系数目为4.

分析

---

对于给定的m中关系，要分析其是否合理。首先要用并查集存储题目给出的关系，如果把一个集合看作是一个朋友圈，那么根据题目要求，一个朋友圈内的任意两个人一定互为朋友关系（这句话很重要），那么，这个朋友圈以2个人为基数，一共有n*(n-1)/2对朋友，即关系数目。那么如果把所有的朋友圈的关系数目加起来，就是n个人一共的关系数目，至于有几个朋友圈，要看题目出入的要求。
使用并查集存储数据，把有朋友关系的元素，归并到一个集合，遍历每一个集合，把所有集合的关系数目加起来，再与m比较即可。
如果了解并查集的知识，此题非常容易。
要注意：当n过大时，n(n-1)/2可能会超出int类型的范围*

ac代码

---

#include <iostream>

using namespace std;

const int MAXN = 150000 + 10;
const int MOD = 10000;
const double eps = 1e-9;

int par[MAXN];
int rak[MAXN];
int n;
long long m;

void init()
{
    for (int i = 0; i < n; i ++ ) {
        par[i] = i;
        rak[i] = 1;
    }
}

int find(int x) {
    if ( par[x] == x ) return x;
    else return par[x] = find(par[x]);
}

void unite(int x, int y) {
    x = find(x);
    y = find(y);
    if ( x == y ) return ;
    if ( rak[x] < rak[y] ) {
        par[x] = y;
        rak[y] += rak[x];
    } else {
        par[y] = x;
        rak[x] += rak[y];
    }
}

int main()
{
    cin >> n >> m;
    init();
    for (int i = 0; i < m; i ++ ) {
        int a, b;
        scanf("%d %d", &a, &b);
        unite(a, b);
    }

    long long ans = 0;
    for (int i = 0; i < n; i ++ ) {
        if ( par[i] == i && rak[i] != 1 ) {

            ans += (long long)(rak[i]) * (rak[i] - 1) / 2;
        }
    }

    if ( ans == m ) printf("YES\n");
    else printf("NO\n");

}