hdu 1325 Is It A Tree?

ACder_chen

于 2014-03-22 19:50:58 发布

阅读量578

点赞数

CC 4.0 BY-SA版权

分类专栏：并查集

本文链接：https://blog.youkuaiyun.com/qqchenjunwei/article/details/21820153

并查集专栏收录该内容

17 篇文章

订阅专栏

本文介绍了一个用于判断一组边是否能够构成一棵树的算法。通过并查集实现节点间的连接关系判断，确保每个节点的入度为1且仅有一个根节点，以此来验证输入的边集合能否构成树。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

#include <iostream>
#include <cstring>
#include <cstdio>


using namespace std;

const int M = 1005;
int vist[M], Set[M];
int du[M];
int a, b, maxn;
int flag, cont, k;
/*
形成树的条件是， 每一个节点的入度只能为1， 
且根只能有一个。
*/

int Find(int x) {

    return x == Set[x]? x : Set[x] = Find(Set[x]);
}

void Union(int x, int y) {

    x = Find(x);
    y = Find(y);
    if(x != y) {

        Set[y] = x;
    }
}

void init() {

    for(int i = 1; i < M; i++) {

        Set[i] = i;
        vist[i] = 0;
        du[i] = 0;
    }
    flag = 1;
    cont = 0;
    k = 0;
    maxn = 0;
}


int main()
{
    int q = 1;
     init();
    while(scanf("%d%d", &a, &b) != EOF) {

        if(a < 0 && b < 0)
            break;
        if(a == 0 && b == 0) {

            for(int i = 1; i < M; i++) {

                if(vist[i] && Set[i] == i) {

                    cont++;
                }
                if(du[i] > 1) {

                    flag = 0;
                    break;
                }

            }
            if(cont >1)
                 flag = 0;
            if(flag)
                 printf("Case %d is a tree.\n", q++);
            else
                printf("Case %d is not a tree.\n", q++);
            flag = 0;
            init();
            continue;
        }
        if(a != b && Find(a) == Find(b)) {

            flag = 0;
        }
        else {

            vist[a] = 1;
            vist[b] = 1;
            du[b]++;
            Union(a, b);
        }


    }
    return 0;
}