nyoj--129 树的判定(并查集)

nyoj 129

题意

若干条有向边$(u, v)$，判定能否构成一棵有向树。

题解

构成有向树的条件是：
1. 空树是有向树；
2. 只有一个根节点，除根节点外的每个结点只有一个父节点，也就是只有一个结点入度为0，其他结点入度为1；
3. 不能形成环。

可以用并查集检查集合个数，显然一棵树应只有一个集合。
一开始没注意到空树，一直WA，WA，WA，伤得不轻。

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

const int maxn = 10000 + 10;
int p[maxn], vis[maxn];

int  find(int x) { return x == p[x] ? x : p[x] = find(p[x]); }

int main()
{

    int a, b;
    int cnt = 1;

    while(scanf("%d %d", &a, &b) && a != -1)
    {
        bool flag = true;
        if(a == 0 && b == 0){   //empty tree
            printf("Case %d is a tree.\n", cnt++);
            continue;
        }

        memset(vis, 0, sizeof(vis));
        for(int i = 0; i < maxn; ++i) p[i] = i;

        if(flag && p[b] == b && find(a) != b) p[b] = a;
        else flag = false;
        vis[a] = vis[b] = true;

        while(scanf("%d %d", &a, &b) && (a || b)){
            if(flag && p[b] == b && find(a) != b) p[b] = a;
            else flag = false;
            vis[a] = vis[b] = true;
        }

        int s = 0;
        for(int i = 0; i < maxn; ++i) if(vis[i] && i == p[i]) s++;
        printf("Case %d is %s tree.\n", cnt++, s == 1 && flag ? "a" : "not a");
    }

    return 0;
}