迷宫城堡

题目描述

为了训练小希的方向感，Gardon建立了一座大城堡，里面有N个房间(N<=10000)和M条通道(M<=100000)，每个通道都是单向的，就是说若称某通道连通了A房间和B房间，只说明可以通过这个通道由A房间到达B房间，但并不说明通过它可以由B房间到达A房间。Gardon需要请你写个程序确认一下是否任意两个房间都是相互连通的，即：对于任意的i和j，至少存在一条路径可以从房间i到房间j，也存在一条路径可以从房间j到房间i。 

输入描述:

输入包含多组数据，输入的第一行有两个数：N和M，接下来的M行每行有两个数a和b，表示了一条通道可以从A房间来到B房间。文件最后以两个0结束。
       

输出描述:

对于输入的每组数据，如果任意两个房间都是相互连接的，输出"Yes"，否则输出"No"。
       

示例1

输入

3 3
1 2
2 3
3 1
3 3
1 2
2 3
3 2
0 0

输出

Yes
No

#include<stdio.h>
#include<string.h>
int pre[2][100010];
int n,m;
int find(int a,int i)
{
    int r=a;
    while(r!=pre[i][r])
    {
        r=pre[i][r];
    }
    return r;
}
void fun(int a,int b)
{   
    if(a!=n)
    {
        int fa=find(a,0),fb=find(b,0);
        if(fa!=fb)
            pre[0][a]=b;
    }
    if(b!=n)
    {
        int fa=find(a,1),fb=find(b,1);
        if(fa!=fb)
            pre[1][b]=a;
    }
}
int main()
{
    while(scanf("%d%d",&n,&m)!=EOF,n||m)
    {
        int i,w=1;
        for(i=0;i<=n;i++)
        {
            pre[0][i]=pre[1][i]=i;
        }
        while(m--)
        {
            int a,b;
            scanf("%d%d",&a,&b);
            fun(a,b);
        }
        for(i=1;i<=n;i++)
        {
            if(find(i,0)!=n||find(i,1)!=n)
            {
                w=0;
                break;
            }
        }
        if(w)
            printf("Yes\n");
        else
            printf("No\n");
    }
}