poj1308&&openjudge310:Is It A Tree?

这个题用并查集判断有没有森林就行了，注意一些细节的特特判，代码写的并不优美，刚开始思路有点乱，导致写了很多啰嗦的东西，然后就懒得精简了，-_-|||

#include <cstdio>
#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;
int fa[500000];
int in[500000];
bool flag = 0;
bool vis[5000000];
int find(int x)
{
    return fa[x] == x ? x : fa[x] = find(fa[x]);
}
int main()
{
    int x,y;int tot = 1;
    while(cin >>x>>y)
    {   
        int a,b;
        if(x == -1 && y == -1)
        break;
        else
        {
            memset(in,0,sizeof(in));
            memset(vis,0,sizeof(vis));
            memset(fa,0,sizeof(fa));
            if(x == 0 && y == 0)
            {
                printf("Case %d is a tree.\n",tot++);
                continue;
            }
            fa[y] = x;
            in[y] ++;
            vis[y] = 1;
            vis[x] = 1;
            while(cin>>a>>b)
            {           
                flag = 0;
                int x;
                int ans = 0;
                if(a == 0 && b == 0)
                {
                    for(int i = 1; i < 10000; i ++)
                    {
                        if(in[i] > 1)
                        {
                            flag = 1;
                        }
                        if(in[i] == 0 && vis[i])
                        {
                            x = i;
                            ans ++;
                        }
                    }
                    if(ans > 1 || ans == 0)
                    flag = 1;
                    else
                    for(int i = 1; i < 1000 ; i ++)
                    {
                        if(find(i) != find(x) && vis[i])
                        {
                            flag = 1;
                        }
                    }
                    if(flag)
                    printf("Case %d is not a tree.\n",tot++);
                    else
                    printf("Case %d is a tree.\n",tot++);
                    memset(in,0,sizeof(in));
                    memset(vis,0,sizeof(vis));
                    memset(fa,0,sizeof(fa));
                    break;
                }
                fa[b] = a;
                in[b]++;
                vis[b] = 1;
                vis[a] = 1;
                if(a == b)
                flag = 1;   
            }
        }
    }
    return 0;
}