POJ 3648 Wedding（2-SAT）

经典2-SAT题目，之前做的一场比赛里面有一道2-SAT的题，当时没什么思路，之后看题解就想学一下2-SAT，今天终于把这个模版题做出来了。感觉2-SAT还是图论里面综合性比较强的题目，需要用到tarjan缩点和拓扑排序，建图的时候也挺纠结的。这题的大意就是说一些夫妻去参加聚会，其中一些人两两之间可能有不正常关系（比如搞基，百合什么的），同一对夫妻不能坐在同一边，有特殊关系的两个人也不能在同一边。基本的思路就是把读入的两个点，第一个点连向第二个点的互斥点，第二个点连向第一个点的互斥点。但是最后还要从新娘向新郎连一条边，这让我百思不得其解啊，问了好几个人也没搞太明白，只是大概知道这是为了避免新郎和新娘同在一边的情况。唯一能想通的解释就是新娘没有去和别人乱搞，所以没有边是从新娘向外连的，只可能从新郎连向新娘，如果出现了这种情况那么从新娘再向新郎连一条边就使这种情况让新郎和新娘之间强连通，判定为错误的情况。这种解释很牵强，求路过的大牛讲一下正确的理解方式……剩下的工作就是套模版啦。

#include <iostream>
#include <cstdio>
#include <string.h>
#include <vector>
#include <map>
#include <stack>
using namespace std;
#define  M 5010              //题目中可能的最大点数
stack<int> s;          //tarjan 算法中的栈
bool ins[M];             //检查是否在栈中
int dfn[M];                  //深度优先搜索访问次序
int low[M];                  //能追溯到的最早的次序
int scnt=0;        //有向图强连通分量个数
int cnt=0;                 //索引号
vector<int> g[M],g2[M];        //邻接表表示
int belong[M];   //记录每个点在第几号强连通分量里
int ant[M];       //缩点后互斥的点
int color[M];    //用于拓扑排序的染色
int ind[M];     //统计入度
int n;           //图的规模
map<char,int> pos;      //简化对h和w的处理
void tarjan(int i)   //缩点求强连通分量
{
    int j;
    dfn[i]=low[i]=cnt++;
    ins[i]=true;
    s.push(i);
    for (int e=0; e<g[i].size(); e++)
    {
        j=g[i][e];
        if (dfn[j]==-1)
        {
            tarjan(j);
            low[i]=min(low[i],low[j]);
        }
        else if (ins[j])
            low[i]=min(low[i],dfn[j]);
    }
    if (dfn[i]==low[i])
    {
        scnt++;
        do
        {
            j=s.top();
            s.pop();
            ins[j]=false;
            belong[j]=scnt;
        }
        while (j!=i);
    }
}
void toposort()   //拓扑排序建反图
{
    int count=0;
    for(int i=1; i<=scnt; i++)
    {
        if(ind[i]==0)
        {
            s.push(i);
            count++;
        }
    }
    while(!s.empty())
    {
        int u=s.top();
        s.pop();
        if(!color[u])
        {
            color[u]=1;  //给两类节点分别染色
            color[ant[u]]=2;
        }
        for(int i=0; i<g2[u].size(); i++)
        {
            ind[g2[u][i]]--;
            if(!ind[g2[u][i]])
                s.push(g2[u][i]);
        }
    }
}
void solve()     //判断是否有解并输出
{
    memset(ins,0,sizeof(ins));
    memset(dfn,-1,sizeof(dfn));
    memset(low,-1,sizeof(low));
    memset(ind,0,sizeof(ind));
    memset(color,0,sizeof(color));
    scnt=cnt=0;
    for(int i=0; i<2*n; i++)
        if(dfn[i]==-1)
            tarjan(i);
    for(int i=0; i<n; i++)
        if(belong[i]==belong[i+n])
        {
            printf("bad luck\n");
            return;
        }
        else
        {
            ant[belong[i]]=belong[i+n];
            ant[belong[i+n]]=belong[i];
        }
    for(int i=0; i<2*n; i++)
    {
        for(int j=0; j<g[i].size(); j++)
        {
            if(belong[i]!=belong[g[i][j]])
            {
                g2[belong[g[i][j]]].push_back(belong[i]);
                ind[belong[i]]++;
            }
        }
    }
    toposort();
    bool ok=true;
    for(int i=1; i<2*n; i++)
    {
        if(color[belong[i]]==color[belong[0]])  //输出和新娘颜色一样的节点，即为和新娘在一边的人
        {
            if(!ok)
                printf(" ");
            else ok=false;
            if(i>n)
                printf("%dh",i-n);
            else printf("%dw",i);
        }
    }
}
int main()
{
    //freopen("input.txt","r",stdin);
    pos['h']=1;   //用map简化了对字符的判定
    int m;
    while(scanf("%d%d",&n,&m),n||m)
    {
        for(int i=0; i<2*n; i++)
        {
            g[i].clear();
            g2[i].clear();
        }
        for(int i=0; i<m; i++)
        {
            int x,y;
            char c1,c2;
            scanf("%d%c%d%c",&x,&c1,&y,&c2);
            g[x+pos[c1]*n].push_back(y+(!pos[c2])*n);
            g[y+pos[c2]*n].push_back(x+(!pos[c1])*n);
        }
        g[0].push_back(n);  //从新娘向新郎连一条边，我把这条边反向或者连双向边都WA了……
        solve();
    }
}