ZOJ 3656 Bit Magic (2-SAT)

题目链接：http://acm.zju.edu.cn/onlinejudge/showProblem.do?problemCode=3656

 

题意：b[i][j] 与a[i], a[j]的关系如下。现在已知b，问是否存在对应的a，若存在输出YES，否则输出NO

void calculate(int a[N], int b[N][N]) {
	for (int i = 0; i < N; ++i) {
		for (int j = 0; j < N; ++j) {
			if (i == j) b[i][j] = 0;
			else if (i % 2 == 1 && j % 2 == 1) b[i][j] = a[i] | a[j];
			else if (i % 2 == 0 && j % 2 == 0) b[i][j] = a[i] & a[j];
			else b[i][j] = a[i] ^ a[j];
		}
	}
}

思路：2-SAT 找清楚连边的条件，找出“只能这样连边”和“一定不能这样连边”等关系，不要轻易去掉该连的边，哪怕会它们直接连成一个环

#include <stdio.h> #include <string.h> #include <algorithm> using namespace std; const int maxn = 1010; const int maxm = 1000000; int n, cnt, top, sn, dep; int dfn[maxn], low[maxn], sta[maxn]; int vis[maxn], col[maxn], head[maxn]; struct edge { int to, nxt; } e[maxm]; void init() { cnt = 0; top = sn = dep = 0; memset(dfn, 0, sizeof(dfn)); memset(col, 0, sizeof(col)); memset(vis, 0, sizeof(vis)); memset(head, -1, sizeof(head)); } void add(int s, int t) { e[cnt].to = t; e[cnt].nxt = head[s]; head[s] = cnt++; } void tarjan(int u) { dfn[u] = low[u] = ++dep; sta[++top] = u; vis[u] = 1; int i, k; for (i = head[u]; ~i; i = e[i].nxt) { k = e[i].to; if(dfn[k] == 0) { tarjan(k); low[u] = min(low[u], low[k]); } else if(vis[k]) low[u] = min(low[u], dfn[k]); } if(dfn[u] == low[u]) { sn++; do { k = sta[top--]; vis[k] = 0; col[k] = sn; } while(k != u); } } bool ok() { for(int i = 0; i < 2 * n; i++) if(!dfn[i]) tarjan(i); for(int i = 0; i < n; i++) if(col[i * 2 + 1] == col[i * 2]) { return false; } return true; } int b[maxn >> 1][maxn >> 1]; int main() { while(~scanf("%d", &n)) { bool flag = true; for(int i = 0; i < n; i++) for(int j = 0; j < n; j++) scanf("%d", &b[i][j]); for(int i = 0; i < n; i++) if(b[i][i] != 0) { flag = false; goto hehe; } for(int k = 0; k < 32; k++) { init(); for(int i = 0; i < n; i++) for(int j = 0; j < n; j++) { if(i == j) continue; int w = b[i][j] & (1 << k); if(i % 2 == 1 && j % 2 == 1) { if(w) { add(i * 2 + 1, j * 2); add(j * 2 + 1, i * 2); } else { add(i * 2, i * 2 + 1); add(j * 2, j * 2 + 1); add(i * 2 + 1, j * 2 + 1); add(j * 2 + 1, i * 2 + 1); } } else if(i % 2 == 0 && j % 2 == 0) { if(w) { add(i * 2 + 1, i * 2); add(j * 2 + 1, j * 2); add(i * 2, j * 2); add(j * 2, i * 2); } else { add(i * 2, j * 2 + 1); add(j * 2, i * 2 + 1); } } else { if(w) { add(i * 2 + 1, j * 2); add(j * 2 + 1, i * 2); add(i * 2, j * 2 + 1); add(j * 2, i * 2 + 1); } else { add(i * 2 + 1, j * 2 + 1); add(j * 2 + 1, i * 2 + 1); add(i * 2, j * 2); add(j * 2, i * 2); } } } if(!ok()) { flag = false; goto hehe; } } hehe: printf("%s\n", flag ? "YES" : "NO"); } return 0; }