首页 > 学院 > 开发设计 > 正文

CF 776D The Door Problem TwoSAT 模板题

2019-11-06 06:08:52
字体:
来源:转载
供稿:网友

题目链接:这里 题意:有n道门,m个开关,每道门由m个开关中的两个控制,若门的状态是0(关),那么改变控制这道门的一个开关之后,门的 状态就变为1(开),若门的状态是1(开),那么改变控制这道门的一个开关之后,门的状态就变为0(关)。 现已知n道门的初始状态,以及m个开关分别控制哪些门,问能不能通过按下某些开关,使得全部门的状态都变为1。 解法:看到每道门恰好两个开关控制,想到2-SAT解决。对于每道门,已知控制这道门的开关是x,y,如果这道门的初始状态为1,那么需要控制它的开关同时为1或者同时开关,相反如果这道门的初始状态为0,那么需要控制它的开关一个开一个关。所以按照这个逻辑建边跑TwoSAT就可以了。这也是我写的第一个TwoSAT,模板是白书模板,感觉也不错的。

#include <bits/stdc++.h>using namespace std;const int maxn = 1e5+5;int door[maxn];vector <int> v[maxn];struct TwoSAT{ int n; vector <int> G[maxn*2]; bool mark[maxn*2]; int S[maxn*2], c; bool dfs(int x){ if(mark[x^1]) return false; if(mark[x]) return true; mark[x] = true; S[c++] = x; for(int i = 0; i < G[x].size(); i++){ if(!dfs(G[x][i])) return false; } return true; } void init(int n){ this->n = n; for(int i = 0; i < n*2; i++) G[i].clear(); memset(mark, 0, sizeof(mark)); } void addedge(int x, int xval, int y, int yval){ x = x * 2 + xval; y = y * 2 + yval; G[x^1].push_back(y); G[y^1].push_back(x); } bool solve(){ for(int i = 0; i < 2*n; i += 2) if(!mark[i] && !mark[i+1]){ c = 0; if(!dfs(i)){//已经尝试过把i标记为true,然而不行,那么在尝试将i+1标记为真时,先把i的影响消掉 while(c > 0) mark[S[--c]] = false; if(!dfs(i + 1)) return false; } } return true; }}twosat;int main(){ int n, m, x, y; cin >> n >> m; for(int i = 0; i < n; i++){ scanf("%d", &door[i]);} for(int i = 0; i < m; i++){ scanf("%d", &x); for(int j = 0; j < x; j++){ scanf("%d", &y); v[y - 1].push_back(i); } } twosat.init(m); for(int i = 0; i < n; i++){ if(door[i]){ twosat.addedge(v[i][0], 0, v[i][1], 1); twosat.addedge(v[i][0], 1, v[i][1], 0); } else{ twosat.addedge(v[i][0], 0, v[i][1], 0); twosat.addedge(v[i][0], 1, v[i][1], 1); } } puts(twosat.solve() ? "YES" : "NO"); return 0;}
发表评论 共有条评论
用户名: 密码:
验证码: 匿名发表