hdu1325 Is It A Tree?

题意:有向图判断是不是一棵树

分析:你管他有向图还是无向图呢,什么吐跟并查集有关系吗。甭管他,分析完提后裸并查集.

#include <iostream>
#include<bits/stdc++.h>
#include<string.h>
#include<math.h>
#define N 11000
 
using namespace std;
 
int pre[N],vis[N],flag,maxn;
 
void init()
{
    for(int i=0;i<N;i++) pre[i]=i;
    memset(vis,0,sizeof(vis));
    maxn=0;
    flag=1;
}
 
int find(int v)
{
    int t1,t2=v;
    while(v!=pre[v]) v=pre[v];
    while(t2!=v)
    {
        t1=pre[t2];
        pre[t2]=v;
        t2=t1;
    }
    return v;
}
 
void unite(int u,int v)
{
    int t1=find(u);
    int t2=find(v);

    //V原先是一棵树的子节点,即V的入度大于1 或 V成有向回路环了 
    if(v!=t2||t1==t2)   flag=0;

    if(t1!=t2)  pre[t2]=t1;
}
 
int main()
{
    int kase=0;
    int u,v;
    init();
    while(~scanf("%d%d",&u,&v)&&u>=0&&v>=0)
    {
        if(u==0&&v==0)
        {
            int tp=0;

            //if判断是不是只有一棵树 
            for(int i=1;i<=maxn;i++)
                if(vis[i]&&i==pre[i]) tp++;

            if(tp<=1&&flag)
                printf("Case %d is a tree.\n",++kase);
            else
                printf("Case %d is not a tree.\n",++kase);
            init();
            continue;
        }
        maxn=max(maxn,max(u,v));
        vis[u]=vis[v]=1;
        unite(u,v);
    }
    return 0;
}

copy的:弱菜zc大佬的,相当硬核的一位大佬.

发布了183 篇原创文章 · 获赞 31 · 访问量 3万+

猜你喜欢

转载自blog.csdn.net/mlm5678/article/details/104622493