POJ 4026 Eve

题目链接：http://livearchive.onlinejudge.org/external/59/5914.pdf

分析：并查集。相同祖先的放入同一个集合，每个集合可能有一个ID编号，也可能没有。

如果最后存在两个集合的ID为确定值且不相同，那么NO

如果最后所有集合都没有ID编号，或者只剩两个集合，一个有编号，一个没编号，那么POSSIBLY

如果最后所有的集合ID编号相同，或者只剩下一个集合，那么YES

#include <cstdio>
#include <cstring>
#include <cstdlib>

const int MAXN = 200000 + 10;

int p[MAXN];               //记录根节点编号
bool isAlive[MAXN];        //判断是否存活
bool isFemale[MAXN];       //记录是父亲还是母亲
int temp[MAXN];            //记录ID
int n, m, k;

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

int main()
{
    int a, b;
    char ch;
    while ( scanf( "%d", &n ) != EOF )
    {
        for ( int i = 1; i <= n; i++ )
        {
            getchar();
            ch = getchar();
            if ( ch == 'F' ) isFemale[i] = true;
            else isFemale[i] = false;
            p[i] = i;
            isAlive[i] = true;
        }

        scanf( "%d", &m );
        int kk = n;
        for ( int i = 0; i < m; i++ )
        {
            scanf( "%d", &a );
            if ( a < 0 ) isAlive[ -a ] = false;
            else
            {
                ++kk;
                isAlive[kk] = true;
                scanf( "%d %c", &b, &ch );
                if ( ch == 'F' )
                    isFemale[kk] = true;
                else isFemale[kk] = false;

                if ( isFemale[a] ) p[kk] = find(a);
                else               p[kk] = find(b);
            }
        }

        scanf( "%d", &k );

        for ( int i = 1; i <= m + n; i++ ) temp[i] = -1;   //每个节点的ID值初始化为-1，即标记为未知

        for ( int i = 0; i < k; i++ )
        {
            scanf( "%d%d", &a, &b );
            temp[ find(a) ] = b;
        }

        //   for ( int i = 1; i <= kk; i++ )
        //        printf( "fa[%d] = %d, temp[%d] = %d\n", i, p[i], i, temp[i] );

        int flag = 0;
        bool first = false;   //是否是第一次进入循环
        int x, y;
        for ( int i = 1; i <= kk; i++ )
            if ( isAlive[i] )
            {
                x = find(i);
             //   printf( "temp[%d] = %d\n", x, temp[x] );
                if ( first )
                {
                    if ( x != y )         //若根节点不同
                    {
                        if ( temp[x] != temp[y] )     //若ID不同
                        {
                            if ( temp[x] != -1 && temp[y] != -1 )   //若存在一组，ID都是确定值
                            {
                                flag = 1;    //那么存活下来的线粒体DNA肯定不同
                                break;
                            }
                        }
                        if ( temp[x] == -1 || temp[y] == -1 )   //若存在一个不确定
                        {
                            flag = -1;
                        }
                    }
                }
                if ( temp[x] == -1 && first ) y = y;
                else y = x;
                first = true;
            }

        if ( flag == 1 ) puts("NO");
        else if ( flag == 0 ) puts("YES");
        else puts("POSSIBLY");
    }
    return 0;
}