CodeForces 776D The Door Problem【并查集】

题目链接：http://codeforces.com/contest/776/problem/D
题意：给你n个门，和m组开关，每扇门都有两个开关控制，每个开关控制x扇门，如果选择了某组开关，则使这组开关里的每个开关控制的所有的门按状态取反，问你是否能使得所有的门状态为1
解析：将每个开关拆分成两个点，选这个开关和不选这个开关（x,x+m），根据每扇门的状态来，如果状态为1，则需要同时选择这两个开关或者不选，如果状态为零，那么只能选择一个开关。根据上面的情况做并查集，然后判断第i（选）组开关和i+m（选）是否联通，有则无解

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cmath>
#include <iostream>
#include <vector>
#include <queue>
#include <string>
#include <set>
#include <stack>
#include <map>
using namespace std;
const int maxn = 2e5+100;
int fa[maxn];
int a[maxn];
int vis[maxn][3];
void init(int n)
{
    for(int i=0;i<=n;i++)
        fa[i] = i;
}
int getfa(int x)
{
    if(x==fa[x])
        return x;
    else
        return fa[x] = getfa(fa[x]);
}
void merge(int u,int v)
{
    int t1 = getfa(u);
    int t2 = getfa(v);
    if(t1!=t2)
        fa[t1] = t2;
}
int main()
{
    int n,m;
    scanf("%d %d",&n,&m);
    init(2*m);
    for(int i=1;i<=n;i++)
        scanf("%d",&a[i]);
    for(int i=1;i<=m;i++)
    {
        int x;
        scanf("%d",&x);
        for(int j=0;j<x;j++)
        {
            int tmp;
            scanf("%d",&tmp);
            vis[tmp][++vis[tmp][0]] = i;
        }
    }
    for(int i=1;i<=n;i++)
    {
        if(a[i])
        {
            merge(vis[i][1],vis[i][2]);
            merge(vis[i][1]+m,vis[i][2]+m);
        }
        else
        {
            merge(vis[i][1],vis[i][2]+m);
            merge(vis[i][1]+m,vis[i][2]);
        }
    }
    int flag = 1;
    for(int i=1;i<=m;i++)
    {
        if(getfa(i)==getfa(i+m))
        {
            flag = 0;
            break;
        }
    }
    if(flag)
        puts("YES");
    else
        puts("NO");
    return 0;
}