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（选）是否联通，有则无解

 1 #include <cstdio>
 2 using namespace std ; 
 3 
 4 const int maxn = 100011 ; 
 5 int n,m,k ; 
 6 int light[ maxn ],g[maxn][3],f[maxn*2] ; 
 7 
 8 inline int getfather(int x) 
 9 {
10     if(x==f[x]) return x ; 
11     f[x] = getfather(f[x]) ; 
12     return f[x] ; 
13 }
14 
15 inline void merge(int x,int y) 
16 {
17     int xx,yy ; 
18     xx = getfather(x) ;   yy = getfather(y) ; 
19     if(xx!=yy) f[xx] = yy ;  
20 }
21 
22 int main() 
23 {
24     scanf("%d%d",&n,&m) ;    
25     for(int i=1;i<=n;i++) 
26         scanf("%d",&light[ i ]) ; 
27     int x,y ; 
28     for(int i=1;i<=m;i++) 
29     {
30         scanf("%d",&k) ; 
31         for(int j=1;j<=k;j++) 
32             scanf("%d",&x) , g[x][++g[x][0]] = i ;  
33     }
34     
35     for(int i=1;i<=2*m;i++) f[ i ] = i ; 
36     for(int i=1;i<=n;i++) 
37     {
38         x = g[i][1] ; 
39         y = g[i][2] ; 
40         if(light[ i ]) 
41         {
42             merge(x,y) ; 
43             merge(x+m,y+m) ; 
44         }
45         else
46         {
47             merge(x,y+m) ; 
48             merge(y,x+m) ; 
49         }
50     }
51     
52     bool flag = 0 ;
53     int xx,yy ;  
54     for(int i=1;i<=m;i++) 
55     {
56         xx = getfather( i ) ; 
57         yy = getfather( i+m ) ;  
58         if(xx==yy) flag = 1 ; 
59     }
60     if(flag) 
61         printf("NO
") ; 
62     else 
63         printf("YES
") ; 
64     return 0 ; 
65 }