CodeForces 776D The Door Problem【并查集】

CodeForces 776D The Door Problem【并查集】
并查集
设 f 1--m 表示 开的情况 m+1--2*m 表示关的情况
对于每盏灯
如果他 是关的 则 x--y x+m--y+m 表示要同关 或者同开
如果他 是开的 则 x+m--y x--y+m 表示一个关 一个开
如果一盏灯 的 x 连向 了 x+m 则表示是矛盾了 那么久是错误的

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

#include <cstdio>
using namespace std ; 

const int maxn = 100011 ; 
int n,m,k ; 
int light[ maxn ],g[maxn][3],f[maxn*2] ; 

inline int getfather(int x) 
{
    if(x==f[x]) return x ; 
    f[x] = getfather(f[x]) ; 
    return f[x] ; 
}

inline void merge(int x,int y) 
{
    int xx,yy ; 
    xx = getfather(x) ;   yy = getfather(y) ; 
    if(xx!=yy) f[xx] = yy ;  
}

int main() 
{
    scanf("%d%d",&n,&m) ;    
    for(int i=1;i<=n;i++) 
        scanf("%d",&light[ i ]) ; 
    int x,y ; 
    for(int i=1;i<=m;i++) 
    {
        scanf("%d",&k) ; 
        for(int j=1;j<=k;j++) 
            scanf("%d",&x) , g[x][++g[x][0]] = i ;  
    }
    
    for(int i=1;i<=2*m;i++) f[ i ] = i ; 
    for(int i=1;i<=n;i++) 
    {
        x = g[i][1] ; 
        y = g[i][2] ; 
        if(light[ i ]) 
        {
            merge(x,y) ; 
            merge(x+m,y+m) ; 
        }
        else
        {
            merge(x,y+m) ; 
            merge(y,x+m) ; 
        }
    }
    
    bool flag = 0 ;
    int xx,yy ;  
    for(int i=1;i<=m;i++) 
    {
        xx = getfather( i ) ; 
        yy = getfather( i+m ) ;  
        if(xx==yy) flag = 1 ; 
    }
    if(flag) 
        printf("NO
") ; 
    else 
        printf("YES
") ; 
    return 0 ; 
}