POJ3678 (Katu Puzzle)[2-SAT]

题目链接:http://poj.org/problem?id=3678

2-SAT模板题,注意四种条件中a & b = 1和a | b = 0的情况,前者需要u^1 -> u, v^1 -> v连边,后者需要u -> u^1, v -> v^1连边。
相当于无论如何都要选u^1(无论如何都要选v^1)。

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
const int MAXN=2009;
const int MAXM=2000009;
struct Edge {
int to,next;
}edge[MAXM];
int head[MAXN],tot;
void init() {
tot=0;
memset(head,-1,sizeof(head));
}
void addedge(int u,int v) {
edge[tot].to=v;edge[tot].next=head[u];head[u]=tot++;
}
bool vis[MAXN];
int S[MAXN],top;
bool dfs(int u) {
if(vis[u^1]) return false;
if(vis[u]) return true;
vis[u]=true;
S[top++]=u;
for(int i=head[u];~i;i=edge[i].next)
if(!dfs(edge[i].to))
return false;
return true;
}
char s[10];

bool Twosat(int n) {
memset(vis,0,sizeof(vis));
for(int i=0;i<n;i+=2) {
if(vis[i] || vis[i^1]) continue;
top=0;
if(!dfs(i)) {
while(top) vis[S[--top]]=false;
if(!dfs(i^1)) return false;
}
}
return true;
}

int main() {
int n,m;
int a,b,c;
int u,v;
scanf("%d%d",&n,&m);
init();
for(int i=0;i<m;i++) {
scanf("%d%d%d",&a,&b,&c);
u=2*a,v=2*b;
scanf("%s",s);
if(s[0]=='A') {
if(c==1) {
addedge(u,u^1);
addedge(v,v^1);
addedge(u^1,v^1);
addedge(v^1,u^1);
}
else {
addedge(u^1,v);
addedge(v^1,u);
}
}
if(s[0]=='O') {
if(c==0) {
addedge(u^1,u);
addedge(v^1,v);
addedge(u,v);
addedge(v,u);
}
else {
addedge(u,v^1);
addedge(v,u^1);
}
}
if(s[0]=='X') {
if(c==0) {
addedge(u,v);
addedge(v,u);
addedge(u^1,v^1);
addedge(v^1,u^1);
}
else {
addedge(u,v^1);
addedge(v^1,u);
addedge(u^1,v);
addedge(v,u^1);
}
}
}
if(Twosat(2*n)) printf("YES");
else printf("NO");
}