POJ1308 Is It A Tree?

题目大意：和HDU1272-小希的迷宫题目一样，

如果有一个通道连通了房间A和B，那么既可以通过它从房间A走到房间B，也可以通过它从房间B走到房间A，为了提高难度，小希希望任意两个房间有且仅有一条路径可以相通（除非走了回头路）。小希现在把她的设计图给你，让你帮忙判断她的设计图是否符合她的设计思路。比如下面的例子，前两个是符合条件的，但是最后一个却有两种方法从5到达8。 

Input

输入包含多组数据，每组数据是一个以0 0结尾的整数对列表，表示了一条通道连接的两个房间的编号。房间的编号至少为1，且不超过100000。每两组数据之间有一个空行。 
整个文件以两个-1结尾。

题目思路：我们想一想，当从一个房间到达另一个房间存在不止一条路时会发生什么，首先肯定存在一个包含这两个点的环，既然存在环，那么这两点的根节点应该相同。好了问题解决了，只要发现新读入的两个相连的房间的根节点相同则证明，着不是个树，标记一下。思路有了，代码就是并查集模板了~

#include<cstdio>
#include<cstdlib>
#include<cmath>
#include<iostream>
#include<algorithm>
#include<cstring>
#include<vector>
#include<queue>
#define INF 0x3f3f3f3f
#define MAX 1000005

using namespace std;

int father[MAX],vis[MAX];

int Find(int x)
{
    while(x!=father[x])
    {
        x=father[x];
    }
    return x;
}

int main()
{
    int a,b,ok,i,k,cnt=1;

    while(scanf("%d%d",&a,&b),a!=-1 || b!=-1)
    {
        k=0;
        memset(vis,0,sizeof(vis));

        vis[a]=1;
        vis[b]=1;

        ok=1;

        if(!a && !b)
        {
            printf("Case %d is a tree.\n",cnt++);
            continue;
        }

        for(i=0; i<MAX; i++)
            father[i]=i;

        int x=Find(a);
        int y=Find(b);

        if(x!=y)
        {
            father[x]=y;
        }

        else
            ok=0;

        while(scanf("%d%d",&a,&b),a||b)
        {
            vis[a]=1;
            vis[b]=1;
            int x=Find(a);
            int y=Find(b);

            if(x!=y)
            {
                father[x]=y;
            }
            else
                ok=0;//标记一下
        }

        if(ok)//当ok=0时就不进入了，节省下时间
        {
            for(i=0;i<MAX;i++)
            {
                if(vis[i] && father[i]==i)
                    k++;
            }
        }

        if(k>1)
            ok=0;

        if(ok)
            printf("Case %d is a tree.\n",cnt++);
        else
            printf("Case %d is not a tree.\n",cnt++);
    }

    return 0;
}