Codeforces 776D：The Door Problem（DFS染色）

http://codeforces.com/problemset/problem/776/D

题意：有n个门，m个开关，每个门有一个当前的状态（0表示关闭，1表示打开），每个开关控制k个门，但是每个门确切的受两个开关控制，如果一个开关打开，那么原来关闭的门会打开，打开的门关闭，问是否存在一个情况使得所有的门打开。

思路：类似于01染色，把开关当成点，门当前的状态当成边权建图。初始先假设一个门的状态（初始假设为0和假设为1都是一样的，举几个例子就发现了），然后因为每个门受两个开关控制，所以可以推出下一个门的状态，如果发现互斥的情况，那么就是“NO”。

这里设成门状态0表示打开，1表示关闭，方便运算。

#include <bits/stdc++.h>
using namespace std;
#define N 200010
struct Edge {
    int v, nxt, w;
} edge[N*2];
int col[N], fir[N], sec[N], head[N], tot, w[N];

void Add(int u, int v, int w) {
    edge[tot] = (Edge) {v, head[u], w}; head[u] = tot++;
    edge[tot] = (Edge) {u, head[v], w}; head[v] = tot++;
}

bool dfs(int u) {
    for(int i = head[u]; ~i; i = edge[i].nxt) {
        int v = edge[i].v, w = edge[i].w;
        if(col[v] == -1) { // 如果这个开关没染色
            col[v] = col[u] ^ w;
            if(!dfs(v)) return false;
        } else if(col[u] ^ col[v] ^ w) return false; // 如果染过色并发生矛盾
    }
    return true;
}

int main() {
    int n, m;
    cin >> n >> m;
    for(int i = 1; i <= n; i++) scanf("%d", &w[i]), w[i] ^= 1;
    for(int i = 1; i <= m; i++) {
        int k; scanf("%d", &k);
        while(k--) {
            int a; scanf("%d", &a);
            if(fir[a] == 0) fir[a] = i; // 连a点的第一个开关
            else sec[a] = i; // 连a点的第二个开关
        }
    }
    memset(head, -1, sizeof(head)); tot = 0;
    for(int i = 1; i <= n; i++)
        Add(fir[i], sec[i], w[i]);
    memset(col, -1, sizeof(col));
    bool flag = 1;
    for(int i = 1; i <= m; i++) {
        if(col[i] == -1) {
            col[i] = 0; // 假设这个开关一开始是关的
            if(!dfs(i)) { flag = 0; break; }
        }
    }
    if(!flag) puts("NO");
    else puts("YES");
    return 0;
}