【codeforces 776D】The Door Problem

【题目链接】:http://codeforces.com/contest/776/problem/D

【题意】

每个门严格由两个开关控制;
按一下开关,这个开关所控制的门都会改变状态;
问你能不能使所有的门都打开(同时);

【题解】

/*    对于每个门,    有两个开关连接着它;    用一条边连接这两个开关    则如果这个门的状态是关的,    则这条边的边权为1    表示它们俩的颜色不能一样;    (也即这两个开关只能改变一个)    也即要1开1关;    如果这个门的状态是开的    则这条边的边权为0；    表示它们俩的颜色相同;    即同时开或同时关.    然后做一个二分图染色    可以就是YES否则NO    也就是说把开关看成节点,门看成边,根据边来确定这条边的两端的节点的颜色是一样还是不同;*/

【完整代码】

#include <iostream>#include <vector>#include <cstdio>#include <cstdlib>#define rei(x) cin >> x#define rep1(i,x,y) for (int i = x;i <= y;i++)using namespace std;const int N = 2e5 + 100;int n, m, a[N],color[N];vector <int> g1[N],g[N],w[N];void dfs(int x, int c){    color[x] = c;    int len = g[x].size();    rep1(i, 0, len - 1)    {        int y = g[x][i];        int ju = w[x][i];        if (color[y] != 0)        {            if (ju == 1 && color[y] == color[x])            {                puts("NO");                exit(0);            }            if (ju == 0 && color[y] != color[x])            {                puts("NO");                exit(0);            }        }        else            if (ju == 0)                dfs(y, c);            else                dfs(y, 3 - c);    }}int main(){    //freopen("D:\\rush.txt", "r", stdin);    rei(n),rei(m);    rep1(i, 1, n)        rei(a[i]);    rep1(i, 1, m)    {        int num;        rei(num);        rep1(j, 1, num)        {            int x;            cin >> x;            g1[x].push_back(n + i);            g1[n + i].push_back(x);        }    }       rep1(i, 1, n)    {        int y1 = g1[i][0], y2 = g1[i][1];        int bian = 1 - a[i];        g[y1].push_back(y2), g[y2].push_back(y1);        w[y1].push_back(bian),w[y2].push_back(bian);            }    rep1(i,n+1,n+m)        if (color[i] == 0)            dfs(i, 1);    puts("YES");    return 0;}