poj 3648 Wedding

http://poj.org/problem?id=3648

新娘和新郎   和其他夫妇 一共n对 每对夫妇不能做在同一侧  而且有奸情的人两个人(男女 男男 女女 出题人口味好重呀)不能同时坐在新娘的对面

2-SAT  

限制条件 n对夫妇一共2×n个人 0--n-1 是新娘  n--2×n-1是对应新郎  染色时 1代表和新娘同侧 -1代表不同侧

首先 新娘必须和新娘同侧2×n-->0 新郎必须和新娘对面n-->n+2×n  

其他夫妇必须异侧

有奸情的有一个在新娘对面 另一个一定和新娘同侧

代码及其注释：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<string>
#include<vector>
#include<map>
#include<queue>
#include<stack>
#include<cmath>
#define LL long long

using namespace std;

const int N=150;
int head1[N],I1;
int head2[N],I2;
struct ss
{
    int j,next;
}side1[N*N],side2[N*N];
int low[N],dfn[N],f[N],deep;
bool in[N],visited[N];
int num[N],sele[N];
stack<int>st;
queue<int>qt;
vector<int>vt[N];
void build1(int x,int y)
{
    side1[I1].j=y;
    side1[I1].next=head1[x];
    head1[x]=I1++;
}
void build2(int x,int y)
{
    side2[I2].j=y;
    side2[I2].next=head2[x];
    head2[x]=I2++;
}
void Tarjan(int x)//将环缩点
{
    visited[x]=true;
    in[x]=true;
    st.push(x);
    low[x]=dfn[x]=deep++;
    for(int t=head1[x];t!=-1;t=side1[t].next)
    {
        int k=side1[t].j;
        if(visited[k]==false)
        {
            Tarjan(k);
            low[x]=min(low[x],low[k]);
        }else if(in[k]==true)
        {
            low[x]=min(low[x],dfn[k]);
        }
    }
    if(low[x]==dfn[x])
    {
        while(st.top()!=x)
        {
            int k=st.top();
            st.pop();
            in[k]=false;
            f[k]=x;
            vt[x].push_back(k);
        }
        int k=st.top();
        st.pop();
        in[k]=false;
        f[k]=x;
    }
}
void Fsearch(int x)//建立新图
{
    for(int t=head1[x];t!=-1;t=side1[t].next)
    {
        int k=side1[t].j;
        if(f[x]!=f[k])
        {
            build2(f[k],f[x]);
            ++num[f[x]];
        }
    }
}
void color(int x,int K)//染色 本环和相对的环 染不同色
{
    x=f[x];
    sele[x]=1;
    for(unsigned int i=0;i<vt[x].size();++i)
    {sele[vt[x][i]]=1;}
    x=(x<K)?x+K:x-K;
    x=f[x];
    sele[x]=-1;
    for(unsigned int i=0;i<vt[x].size();++i)
    {sele[vt[x][i]]=-1;}
}
void subnum(int x)//拓扑 减边
{
    for(int t=head2[x];t!=-1;t=side2[t].next)
    {
        int k=side2[t].j;
        --num[f[k]];
        if(num[f[k]]==0)
        qt.push(f[k]);
    }
}
int main()
{
    //freopen("data.txt","r",stdin);
    int n,m;
    while(scanf("%d %d",&n,&m)!=EOF)
    {
        if(n==0&&m==0)
        break;
        memset(head1,-1,sizeof(head1));
        I1=0;
        build1(0+2*n,0);//新娘和自己同侧
        build1(n,n+2*n);//新郎对面
        for(int i=1;i<n;++i)
        {
            build1(i,i+n+2*n);//其他夫妇 不能同侧
            build1(i+n,i+2*n);
            build1(i+2*n,i+n);
            build1(i+n+2*n,i);
        }
        for(int i=0;i<m;++i)
        {
            int k1,k2;
            char c1,c2;
            scanf("%d%c%d%c",&k1,&c1,&k2,&c2);
            if(c1=='h')
            k1+=n;
            if(c2=='h')
            k2+=n;
            build1(k1+2*n,k2);//有一个在新娘对面 另一个一定在新娘同侧
            build1(k2+2*n,k1);
        }
        while(!st.empty())
        st.pop();
        for(int i=0;i<4*n;++i)
        {vt[i].clear();f[i]=i;}
        memset(in,false,sizeof(in));
        memset(visited,false,sizeof(visited));
        deep=0;
        int l;
        for(l=0;l<4*n;++l)
        {
            if(visited[l]==false)
            Tarjan(l);
            if(l<2*n&&f[l]==f[l+2*n])
            break;
        }
        if(l<2*n)
        printf("bad luck\n");
        else
        {
            memset(head2,-1,sizeof(head2));
            memset(num,0,sizeof(num));
            I2=0;
            for(int i=0;i<4*n;++i)//将缩点 建新边
            Fsearch(i);
            for(int i=0;i<4*n;++i)
            if(f[i]==i&&num[i]==0)//拓扑中 为0的入队列
            qt.push(i);
            memset(sele,0,sizeof(sele));
            while(!qt.empty())
            {
                int k=qt.front();
                qt.pop();
                if(sele[k]==0)//染色
                color(k,2*n);
                subnum(k);
            }
            for(int i=1;i<n;++i)
            {
                if(sele[i]==1)
                printf("%dw",i);
                else
                printf("%dh",i);
                if(i<n-1)
                printf(" ");
                else
                printf("\n");
            }
        }
    }
    return 0;
}