POJ 3648 Wedding（2-SAT输出一组解）

题意：有新郎新娘举办婚礼，吃饭的时候大家坐在一长方形桌，桌子有两列，新娘坐一列上看对面一列，她定义以下事情会让她不开心：夫妻都在对面一列或有仇人在对面一列。问是否能安排出合理方案使得新娘开心。如果有，输出YES并且输出新娘这一列坐着的人，否则输出bad luck。

解法：夫妻不能在同一列，所以夫妻要不在新娘这排要不不在新娘这排，用2-SAT可以很好满足这个条件。然后读入仇人的信息，标记好，用2-sat判断下是否存在解即可。但是他是要输出和新娘同一排的人，那么想到之前做了一道2-sat输出最小字典序的模板，直接套上来用。

代码如下：

#include<iostream>
#include<cstring>
#include<cstdio>
const int maxn = 50000 * 2;
const int maxm = 500100;
using namespace std;
int n, m;
int s[maxn], cnt;
bool mark[maxn];
int head[maxn], to[maxm << 1], nx[maxm << 1], tot;
void addedge(int u, int v) {
    to[tot] = v, nx[tot] = head[u], head[u] = tot++;
}
void init() {
    tot = 0;
    memset(head, -1, sizeof(head));
    memset(mark, 0, sizeof(mark));
}
bool dfs(int u) {
    if(mark[u ^ 1]) 
		return false;
    if(mark[u]) 
		return true;
    s[++cnt] = u;
    mark[u] = 1;
    for(int i = head[u]; ~i; i = nx[i]) {
        if(!dfs(to[i]))
            return false;
    }
    return true;
}
bool twosat() {
	mark[0] = 1;
    for(int i = 1; i <=n-1; i++){
		int u=i*2;
		if(mark[u]||mark[u+1])
			continue;
        cnt=0;
		if(!dfs(u)){
			while(cnt)
				mark[s[cnt--]]=0;
	        if(!dfs(u+1)){
				return false;
			}
		}
	}
	return true;
}

int main() {
    while(scanf("%d%d", &n, &m) && n && m){
        init();
        while(m--) {
        	int u, v;
        	char c1, c2;
            scanf("%d%c%d%c", &u, &c1, &v, &c2);
            u = u * 2 + (c1 == 'h');
            v = v * 2 + (c2 == 'h');
            addedge(u ^ 1, v);
            addedge(v ^ 1, u);
        }
        
        if(twosat()) {
            for(int i = 1; i < n; i++) {
                printf("%d%c ", i, mark[i * 2] ? 'w' : 'h');
            }
            printf("\n");
        } else {
            printf("bad luck\n");
        }
    }
    return 0;
}