POJ 3648-Wedding(2-SAT)
题面很邪恶啊。。。
一对新人请n-1对夫妻吃饭,人们坐在一张桌子的两侧,每一对互为夫妻关系的人必须坐在桌子的两侧。而且有些人两两之间会存在“通奸”关系,通奸关系不仅在男女之间,同性之间也有。新娘对面不可以座有通奸关系的人。判断是否存在可行的排座方案,存在的话输出和新娘同一排的人。
因为新娘对面不可以做有通奸关系的人,也就是说2sat求出的一组可行解是新娘对面的。
如果u和v有通奸关系,就连边u->v',v->u'。
有一点需要注意,就是要连一条边0->1
这样如果选了0就必须选1,那么就矛盾了,所以0一定不被选,选出来的就是新郎那一边的。很巧妙啊!
#include <iostream> #include <cstdio> #include <cstring> #include <algorithm> #include <queue> using namespace std; const int N = 1010; const int M = 100010; struct Edge { int from, to, next; } edge[M], edge2[M]; int head[N]; int cntE, cntE2; void addedge(int u, int v) { edge[cntE].from = u; edge[cntE].to = v; edge[cntE].next = head[u]; head[u] = cntE++; } void addedge2(int u, int v) { edge2[cntE2].from = u; edge2[cntE2].to = v; edge2[cntE2].next = head[u]; head[u] = cntE2++; } int dfn[N], low[N], idx; int stk[N], top; int in[N]; int kind[N], cnt; void tarjan(int u) { dfn[u] = low[u] = ++idx; in[u] = true; stk[++top] = u; for (int i = head[u]; i != -1; i = edge[i].next) { int v = edge[i].to; if (!dfn[v]) tarjan(v), low[u] = min(low[u], low[v]); else if (in[v]) low[u] = min(low[u], dfn[v]); } if (low[u] == dfn[u]) { ++cnt; while (1) { int v = stk[top--]; kind[v] = cnt; in[v] = false; if (v == u) break; } } } int opp[N], ind[N], col[N]; // 相对的点 入度 染色 col[]=1选择 bool topsort(int n) // 序号从0开始 { for (int i = 0; i < 2*n; i += 2) { int k1 = kind[i]; int k2 = kind[i^1]; // 相对的两个的关系 //printf("%d %d %d %d\n", i, i^1, k1, k2); if (k1 == k2) return false; opp[k1] = k2; opp[k2] = k1; } memset(head, -1, sizeof head); int u, v; for (int i = 0; i < cntE; ++i) { u = edge[i].from, v = edge[i].to; if (kind[u] != kind[v]) { // 反向建图 addedge2(kind[v], kind[u]); ind[kind[u]]++; } } queue<int> q; for (int i = 1; i <= cnt; ++i) if (!ind[i]) q.push(i); while (q.size()) { u = q.front(); q.pop(); if (!col[u]) col[u] = 1, col[ opp[u] ] = -1; for (int i = head[u]; i != -1; i = edge2[i].next) if (--ind[edge2[i].to] == 0) q.push(edge2[i].to); } return true; } void init() { cntE = cntE2 = 0; memset(head, -1, sizeof head); memset(dfn, 0, sizeof dfn); memset(in, false, sizeof in); idx = top = cnt = 0; memset(ind, 0, sizeof ind); memset(col, 0, sizeof col); } int main() { int n, m; int u, v; while (scanf("%d%d", &n, &m) == 2) { if (n == 0 && m == 0)break; init(); while (m--) { //3h 7h char s1, s2; scanf("%d%c%d%c", &u, &s1, &v, &s2); u = s1=='w' ? u*2 : u*2+1; v = s2=='w' ? v*2 : v*2+1; if (!u || !v) continue; addedge(u, v^1); addedge(v, u^1); } addedge(0, 1); for (int i = 0; i < 2 * n; ++i) { if (!dfn[i]) tarjan(i); } if (topsort(n)) { for (int i = 1; i < n; i++) { if (col[ kind[2*i] ] == 1) printf("%dh", i); else printf("%dw", i); if (i < n - 1) printf(" "); else printf("\n"); } } else printf("bad luck\n"); } return 0; }