hdu 2545 并查集

题目描述：给定一个无向图，判断这个图是否满足任意两点之间有且仅有一条通路。

思路：并查集，若a和b之间有一条边且处于不同的集合中，则将a和b所在集合合并；若a和b本就在同一集合中（有一条通路），则加上这条a到b便有不止一条通路。另外还需要判断一下是否所有的点最后都能在一个集合中，即根的数量是否等于1.（其实说白了就是判断该图是不是一棵树）

注意：0 0的情况应该输出Yes（空树）

#include <iostream>
#include <cstring>
#include <cstdio>
using namespace std;

const int N = 100007;
int f[N];
bool mark[N];

void init()
{
    for ( int i = 1; i < N; i++ )
    {
        mark[i] = false;
        f[i] = i;
    }
}

int find_f( int x )
{
    if ( f[x] != x ) f[x] = find_f(f[x]);
    return f[x];
}

void union_set( int x, int y )
{
    //f[x] = y居然会栈溢出
    f[y] = x;
}

int main ()
{
    int a, b;
    while ( 1 )
    {
        scanf("%d%d", &a, &b);
        if ( a == -1 && b == -1 ) break;
        if ( a == 0 && b == 0 ) 
        {
            printf("Yes
");
            continue;
        }
        init();
        mark[a] = mark[b] = true;
        union_set( a, b );
        bool flag = true;
        while ( 1 )
        {
            scanf("%d%d", &a, &b);
            if ( a == 0 && b == 0 ) break;
            if ( !flag ) continue;
            mark[a] = mark[b] = true;
            a = find_f(a), b = find_f(b);
            if ( a == b ) flag = false;
            else union_set( a, b );
        }
        if ( !flag )
        {
            printf("No
");
            continue;
        }
        int cnt = 0;
        for ( int i = 1; i < N; i++ )
        {
            if ( mark[i] && f[i] == i ) cnt++;
        }
        if ( cnt == 1 ) printf("Yes
");
        else printf("No
");
    }
    return 0;
}