题意:有向图判断是不是一棵树
分析:你管他有向图还是无向图呢,什么吐跟并查集有关系吗。甭管他,分析完提后裸并查集.
#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大佬的,相当硬核的一位大佬.