G - Bear and Friendship Condition(并查集)

题目链接：CodeForces - 791B
题意：朋友关系，定义如果x和y是朋友，y和z是朋友，那么x和z也使朋友，给出朋友关系图，问你这张图是否正确。

思路：并查集。仔细思考，对于一个满足题意的连通图，假如说一个连通图有n个点那么边的个数应该是n*(n+1)/2。

#include<bits/stdc++.h>
#define N 150010
using namespace std;

typedef long long ll;

int fa[N];

int fin(int x)
{
    if(x == fa[x])
    return x;
    return fa[x] = fin(fa[x]);
}

int main()
{
    int n,m,x,y;
    ll num[N];
    cin >> n >> m;
    memset(num,0,sizeof(num));
    for(int i = 0; i <= n; i++)
    {
        fa[i] = i;
    }
    for(int i = 1; i <= m; i++)
    {
        cin >> x >> y;
        int q = fin(x);
        int p = fin(y);
        fa[q] = p;
    }
    for(int i = 0; i <= n; i++)
    {
        num[fin(i)]++;//这里的意思就是看每一个头结点有多少子节点，当然包括他自己，
    }//所以num数组表示的就是前面所说的一个连通图有多少节点
    ll sum = 0;
    for(int i = 0; i <= n; i++)
    {
        if(num[i] > 1)//假如说这是一个除了他自己以外仍有点和他相连的头结点，那么我
        {//们就先默认这个连通图已经满足题意了
            sum += (num[i]*(num[i]-1))/2;
        }
    }
    if(sum == m)//最后如果边的个数与输入时相等则满足题意
    {
        cout << "YES" << endl;
    }
    else
    {
        cout << "NO" << endl;
    }
    return 0;
}