POJ-3648-Wedding（2-sat）

POJ 2-sat 六题之四

http://blog.sina.com.cn/s/blog_64675f540100k1g9.html

题目描述：

一堆夫妇去参加一对新人的婚礼。人们坐在一个很长很长的桌子的两侧（面对面）。新郎新娘在桌子头面对面座。

新娘不希望看见她对面的一排有一对夫妇坐着（夫妇需要分开两排座）。

同时，一些人之间有暧昧关系，新娘也不希望有暧昧关系的人同时坐在她对面的一排。

问你可否满足新娘的要求，可以的话，输出一种方案。

解题报告：

首先，每个人都可能坐在桌子两侧的某一侧，这样，把每个人拆成两个点。

第一个点表示桌子左侧（我自己定义的），第二个表示右侧。即i和i’

新娘我让她坐在左侧，新郎坐在右侧。

新娘编号0，加入边i’-》i，就是说如果新娘坐右侧的话，就需要坐到左侧（这样就限定了新娘坐在左侧），新郎同样处理。

夫妇一定一左侧，一右侧。

有暧昧关系的人要么同时坐在左侧（不让新娘看见），要么一个坐在左侧，一个坐在右侧。

// File Name: 3648.cpp
// Author: zlbing
// Created Time: 2013/1/31 0:06:20

#include<iostream>
#include<string>
#include<algorithm>
#include<cstdlib>
#include<cstdio>
#include<set>
#include<map>
#include<vector>
#include<cstring>
#include<stack>
using namespace std;
#define MAXN 200
struct TwoSAT{
    int n;
    vector<int>G[MAXN*2];
    bool mark[MAXN*2];
    stack<int>S;
    bool dfs(int x)
    {
        if(mark[x^1])return false;
        if(mark[x])return true;
        mark[x]=true;
        S.push(x);
        for(int i=0;i<G[x].size();i++)
        {
            int v=G[x][i];
            if(!dfs(v))return false;
        }
        return true;
    }
    void init(int _n)
    {
        n=_n;
        for(int i=0;i<2*n;i++)
            G[i].clear();
        memset(mark,0,sizeof(mark));
    }
/*    void add_clause(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);
    }
*/
    void add_clause(int x,int y)
    {
        G[x].push_back(y);
    }
    bool solve()
    {
        for(int i=0;i<2*n;i=i+2)
        {
            if(!mark[i]&&!mark[i+1]){
                  while(!S.empty())
                  {
                      S.pop();
                  }
                if(!dfs(i))
                {
                    while(!S.empty())
                    {
                        mark[S.top()]=false;
                        S.pop();
                    }
                    if(!dfs(i+1))return false;
                }
            }
        }
        char e;
        bool first=true;
        for(int i=4;i<4*n;i++)
            if(mark[i]){
                int k=i/4;
                if(i%4<2)e='w';
                else e='h';
                if(i%2==0){
                    if(first){
                    printf("%d%c",k,e);
                    first=false;        
                    }else 
                    printf(" %d%c",k,e);
                }
            }
        printf("\n");
        return true;
    }
};
TwoSAT solver;
int main(){
    int n,m;
    while(~scanf("%d%d",&n,&m))
    {
        solver.init(2*n);
        if(n==0&&m==0)
            break;
        int a,b;
        char cha,chb;
        solver.add_clause(1,0);
        solver.add_clause(2,3);
        for(int i=1;i<n;i++)
        {
            a=i*2,b=i*2+1;
            solver.add_clause(a*2,b*2+1);
            solver.add_clause(b*2+1,a*2);
            solver.add_clause(a*2+1,b*2);
            solver.add_clause(b*2,a*2+1);
        }
        for(int i=0;i<m;i++)
        {
            scanf("%d%c %d%c",&a,&cha,&b,&chb);
            a=cha=='w'?a*2:a*2+1;
            b=chb=='w'?b*2:b*2+1;
            solver.add_clause(a*2+1,b*2);
            solver.add_clause(b*2+1,a*2);
        }
        if(!solver.solve())
            printf("bad luck\n");
    }
    return 0;
}