HDU1272 小希的迷宫 并查集

/*并查集的应用，注意点的个数为0的时候，注意输入*/
#include <iostream>
#include "UFS.h"
using namespace std;

UFS ufs;
bool flag[MAXN]={0}, f;
int cnt_e, cnt_v;
void init();
int main()
{
int n,m;
init();
while(cin>>n>>m)
{
if(m==-1&&n==-1)return 0;
if(m==0&&n==0)
{
getchar();getchar();
/*条件：1.ufs不冲突2.v=e+1(v=0时例外)*/
if(f) cout<<"No"<<endl;
else if(cnt_v==0)cout<<"Yes"<<endl;
else if(cnt_v!=cnt_e+1)cout<<"No"<<endl;
else cout<<"Yes"<<endl;
init();
}
else
{
if(flag[m] == 0)
{
flag[m] = 1;
cnt_v++;
}
if(flag
== 0)
{
flag
= 1;
cnt_v++;
}
m = ufs.Find(m);
n = ufs.Find(n);
if(m==n) f=1;
else
{
ufs.Union(m, n);
cnt_e++;
}
}
}
return 0;
}
void init()
{
ufs.clear();
memset(flag, 0, sizeof(flag));
f = 0;
cnt_v = 0;
cnt_e = 0;
}