HDU 1272 并查集

/*
    这个是并查集的题目，要考虑的地方有
    1、要保证边的数目比点的数目小1
    2、每个测试数据都是已经给出的，这里用mark数组标记了所有给出的数据
    3、要考虑只输入0 0的情况
    4、当集合数目为1时，才满足条件
*/
#include<iostream>
using namespace std;
const int maxn = 100002;
int vid[maxn], mark[maxn];

int Find(int t)
{
    if(vid[t] != t)
        vid[t] = Find(vid[t]);
    return vid[t];
}

void Merge(int a, int b)
{
    int x, y;
    x = Find(a);
    y = Find(b);
    if(x > y) vid[x] = y;
    else vid[y] = x;
}

int main()
{
    int a, b;
    while(cin >> a >> b)
    {
        if(a==-1 && b==-1) break;
        if(a==0 && b==0)//若只输入0 0的话就符合条件，直接输出Yes
        {
            cout << "Yes" << endl;
            continue;
        }
        int nax = -1, nin = maxn, flag = 0;
        for(int i = 1; i < maxn; i++)//初始化vid数组和mark数组
        {
            vid[i] = i;
            mark[i] = 0;
        }
        while(a || b)
        {
            if(nax < a) nax = a;
            if(nax < b) nax = b;
            if(nin > a) nin = a;
            if(nin > b) nin = b;
            mark[a] = mark[b] = 1;//标记给出的数据
            int aa = Find(a);
            int bb = Find(b);
            if(aa == bb)//若给出的数据已经连接在一起(aa==bb表示已经连接过)，用flag标记
                flag = 1;
            else Merge(a,b);
            cin >> a >> b;
        }
        if(flag)//flag为真表示点的数目可能小于等于边的数目，不符合条件
        {
            cout << "No" << endl;
            continue;
        }
        int cont = 0;
        for(int i = nin; i <= nax; i++)//只对出现过的数据进行判断
            if(mark[i] && vid[i] == i) cont++;
        if(cont == 1) cout << "Yes" << endl;//只有一个集合才满足条件
        else cout << "No" << endl;
    }
    return 0;
}