【并查集】-HDU1272-小希的迷宫

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=1272点击打开链接

题目描述：

设计一个迷宫，迷宫应满足这样的要求：

从任意一点出发到达任意另一点的有且仅有一条。判断输入的迷宫是否符合要求。

解题思路：

并查集查找，解题办法就是在合并函数里判断一下，如果这个点已经被归入集合了，现在又要让它重新归入自己一遍的话就说明我们走重复了，从这点出发有两条路可走。立起一个FLAG 表示就行了。详见代码。另外，要判断这是不是最后只有一个集合，要保证所有地方都能走到。（这点我设了一个标记数组use解决）然后最尼玛坑的是！这道题如果只输入一个“0 0”竟然要输出“Yes”，尼玛在逗我？！（我引入一个num变量，去特判了）

AC代码：

#include <cstdio>
#include <cstring>
#include <iostream>

using namespace std;

int arr[100050],use[100050];
int flag,ans,num;

void reset()
{
    num=0;
    flag=0;
    ans=0;
    memset(arr,0,sizeof(arr));
    memset(use,0,sizeof(use));
    for(int i=1;i<=100010;i++)
        arr[i]=i;
}

int find(int x)
{
    return x==arr[x]?x:find(arr[x]);
}

void mer(int a,int b)
{
    int fa,fb;
    fa=find(a);
    fb=find(b);
    if(fa==fb)
        flag=1;
    else
        arr[fa]=fb;
}

int main()
{
    int from,to,i;
    //freopen("1272_input.txt","r",stdin);
    reset();
    while(scanf("%d%d",&from,&to),from>=0&&to>=0)
    {
        if(from>0&&to>0)
        {
            num++;
            use[from]=1;
            use[to]=1;
            mer(from,to);
        }
        else if(!from&&!to)
        {
            //cout<<"num="<<num<<endl;
            for(i=0;i<100010;i++)
            {
                //if(use[i])cout<<i<<" "<<arr[i]<<endl;
                if(use[i]&&arr[i]==i)ans++;
            }
            //cout<<"flag="<<flag<<endl;
            if(num&&(flag||ans!=1))
                puts("No");
            else
                puts("Yes");
            reset();
        }
    }
    return 0;
}

AC截图：