【判断是不是树】HDU 1272 小希的迷宫

HDU 1272 小希的迷宫

题意：输入多组无向边，结点的标号是随机的（可以不连续也可以不从1开始）。然后让我们判断是不是刚好是一棵树。

需要注意的就是：0 0空树的情况。

代码：（带注释）

#include <iostream>
#include <algorithm>
#include <cstdio>
#include <cmath>
#include <cstring>
#include <string>
#include <queue>
#include <vector>
#include <set>
#include <stack>
#include <list>
#include <map>

using namespace std;
typedef long long ll;
const int maxN = 1e5+5 ;

int root[maxN], high[maxN], size[maxN]; bool vis[maxN];
int flag, cnt = 0, MAX = maxN - 5;

void init()
{
    for(int i= 0; i <= MAX ; i ++)
    {
        root[i] = i;
        high[i] = 0;
        size[i] = 1;
        vis[i] = false;
    }
    flag = 0;
    MAX = 0;
    cnt = 0;
}

int Find(int x) { return root[x] == x ? x : root[x] = Find(root[x]); }
int Same(int x, int y) { return Find(x) == Find(y); }
void Merge(int x, int y)
{
    x = Find(x);
    y = Find(y);
    if(high[x] < high[y] || size[x] < size[y])
    {
        root[x] = y;
        size[y] += size[x] ;
    }
    else
    {
        root[y] = x;
        size[x] += size[y] ;
        if(high[x] == high[y] ) high[x] ++ ;
    }
}

int Num()
{
    set<int>st;
    for(int i = 1 ; i <= MAX ; i ++ )//MAX:最大结点标号
    {
        if(vis[i])//我们将所有结点的祖先结点都给放进set里，如果所有节点的祖先结点是同一个，那么就说明是在同一颗树上
            st.insert(Find(i));
    }
    return st.size();
}

int main()
{
    int a, b;
    init();
    while(~scanf("%d%d", &a, &b) && a != -1 && b != -1)
    {
        cnt ++;
        MAX = max(MAX, a);
        MAX = max(MAX, b);
        if(cnt == 1 && a == 0 && b == 0)//如果是空树也是可以的，cnt的作用就是用来判断空树的，就是0 0的样例
        {
            printf("Yes\n");
            init();
            continue;
        }
        if( a == 0 && b == 0 )
        {
            if(flag || Num() != 1)//必须要先判断flag，如果存在自环我们是没有将后面输入的边连结起来的，所以Num是无意义的
                printf("No\n");
            else
                printf("Yes\n");
            init();
            continue;
        }
        if(!vis[a]) vis[a] = true;
        if(!vis[b]) vis[b] = true;
        if(!flag)//有一个不符合就可以不用走下面的两步了
        {
            if(a == b || Same(a, b) )//自环或者在a，b已经在一颗树上了，就直接是不符合
                flag  = 1;
            else
                Merge(a, b);
        }
        else continue;
    }
    return 0;
}