小希的迷宫

hdoj 1272

题目大意：给出一个图，注意这是个有向图（小希希望任意两个房间有且仅有一条路径可以相通（除非走了回头路）），由于这个错误调试了一上午

解决：与poj 1308 一模一样，用并查集判断图的连通性，统计结点和边的个数来判断是否有回路

#include <iostream>
#include <set>
#include <utility>
using namespace std;
int num[100005];
bool mark[100005];
int Max;
set<pair<int,int> > s;
int find(int x)
{
    int t=x,a;
    while(num[t]>0)t=num[t];
    for(int i=x;num[i]>0; i=num[a])
    {
       a=i;
       num[i]=t;
    }
    return t;
}
void merge(int a,int b)
{
    int fa=find(a);
    int fb=find(b);
    if(fa==fb)return ;
    int t=num[fa]+num[fb];
    if(num[fa]>num[fb]){num[fa]=fb;num[fb]=t;}
    else {num[fb]=fa;num[fa]=t;}
}
void init()
{    
      Max=-1;
      s.clear();
      memset(num,-1,sizeof(num));
      memset(mark,0,sizeof(mark));
}
int main()
{
    int a,b,cnt=0,edge=0,vex=0;
    init();
    while(scanf("%d%d",&a,&b),a>=0 && b>=0)
    {
        if(a==0 && b==0)
        {
            for(int i=1;i<=Max && cnt!=2;i++)
            {
                if( mark[i] )
                {
                    vex++;
                    if(num[i]<0)cnt++;
                }
            }
            edge=s.size();
           //判断0 0的情况
            if(cnt==0 && vex==0 && edge==0){printf("Yes\n");continue;}
            if(cnt==1 && edge==vex-1)printf("Yes\n");
            else printf("No\n");
            cnt=0;
            edge=0;
            vex=0;
            init();
            continue;
        }
        if(a!=b)
        {
             s.insert(make_pair(b,a));  
             merge(a,b);
        }
        if(a>Max)Max=a;
        if(b>Max)Max=b;
        mark[a]=mark[b]=true;
  
    }
 //   system("pause");
    return 0;
}