HDU 1272 小希的迷宫 （并查集）

题目：中文的~~~~

思路：属于并查集算法，输出YES的条件有两个，第一：每次新输入的两个数不能同属于一个集合（即根节点一样），第二：所有的输入完成后判断是否仅有一个集合（根节点只有一个）。只有这两个条件全部达成，才能输出“YES”~~~需要注意的是 直接输入“0 0”，应该输出“YES”。

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <cmath>
#include <cctype>
#include <algorithm>
using namespace std;
const int MAXN = 1e5 + 3;
int pre[MAXN];
bool bol[MAXN];
int maxn = -1;
int flag = 1;   //标记是否出现了根节点一样的两个点、即条件一、默认是没有出现

int Find(int x)
{
    int r = x;
    while(pre[r] != r)
    {
        r = pre[r];
    }
    int i = x,j;
    while(pre[i] != r)
    {
        j = i;
        i = pre[i];
        pre[j] = r;
    }
    return r;
}

void Mix(int a,int b)
{
    int x = Find(a);
    int y = Find(b);
    if(x > y)
    {
        pre[x] = y;
    }
    if(x < y)
    {
        pre[y] = x;
    }
}

void Mst()  //初始化
{
    maxn = -1;
    flag = 1;
    for(int i = 1; i <= MAXN; i++)
    {
        pre[i] = i;
        bol[i] = false;
    }
}

int Search(int maxn) //计算集合的个数
{
    int cnt = 0;
    for(int i = 1; i <= maxn; i++)
    {
        if(bol[i])
            if(pre[i] == i)
                cnt++;
    }
    if(maxn <= 0)
        return 1;
    return cnt;
}

int main()
{
    int M,N;
    Mst();
    while(~scanf("%d%d",&M,&N)&&(M != -1 || N != -1))
    {
        maxn = max(max(M,N),maxn);
        bol[M] = true;     //这个点出现过
        bol[N] = true;
        if(M||N)
        {
            int m = Find(M);
            int n = Find(N);
            if(flag && m == n)    
                flag = 0;       //不满足条件一
            else
                Mix(m,n);
        }
        else
        {
            int single = Search(maxn);    //是否只有一个根节点
            if(single== 1 && flag) printf("Yes\n");    //两个条件都满足
            else printf("No\n");
            Mst();//注意数据初始化
        }
    }
    return 0;
}