POJ3648 2-SAT基础

题目链接
题目有两个限制条件：
1.一对夫妻不能坐在同一边
2.有奸情的两人不能坐在新娘对面（但是可以坐在新娘的同侧）

分点：
有n对夫妻，那就有2n个人，每个人坐在左侧还是右侧又可以分为两种情况，一共是4n个点。i2+1表示第i对的丈夫，i2表示第i对的妻子，在此基础上*2表示在左，*2+1表示在右。

建边
首先对所有的夫妻根据限制1建边，然后根据输入的奸情信息进行限制2的建边。然后判断有没有矛盾即可。

AC代码

//AC
//2021/10/07
#include <iostream>
#include <string.h>
#include <vector>
#define maxn 10001
using namespace std;

vector<int> g[maxn * 2];
bool vis[maxn * 2];
int top = 0, sta[maxn * 2];
inline void addEdge(int u, int dir1, int v, int dir2)
{
    //i*2+1表示第i对的丈夫，i*2表示第i对的妻子
    //在此基础上*2表示在左，*2+1表示在右
    u = u * 2 + dir1;
    v = v * 2 + dir2;
    g[u].push_back(v);
}

void init(int n)
{
    for (int i = 0; i < 2 * n; i++)
        g[i].clear();
    memset(vis, false, sizeof(vis));
    addEdge(0, 1, 0, 0); //新娘在右->新娘在左，矛盾，因此可以保证新娘在左
    addEdge(1, 0, 1, 1); //同理保证新郎在右
    for (int i = 2; i < n; i += 2)
    {
        //增加其他夫妻之间的关系
        addEdge(i, 0, i + 1, 1);
        addEdge(i, 1, i + 1, 0);
        addEdge(i + 1, 0, i, 1);
        addEdge(i + 1, 1, i, 0);
    }
}
bool dfs(int x)
{
    //1^1=0,0^1=1，表示dfs中遇到了一个人既在左侧又在右侧的情况
    if (vis[x ^ 1]) return false;
    if (vis[x]) return true;
    vis[x] = true;
    sta[top++] = x;
    for (int i = 0; i < g[x].size(); i++)
        if (!dfs(g[x][i])) return false;
    return true;
}

bool check(int n)
{
    n <<= 2;
    for (int i = 0; i < n; i += 2)
    {
        if (!vis[i] && !vis[i + 1])
        {
            top = 0;
            if (!dfs(i)) //在左侧的情况出现了矛盾
            {
                while (top > 0)
                    vis[sta[--top]] = false;
                //尝试在右侧的情况，如果还有矛盾就说明没有符合的方案
                if (!dfs(i + 1)) return false;
            }
        }
    }
    return true;
}

int main()
{
    int n, m;
    while (scanf("%d%d", &n, &m) != EOF && (n + m))
    {
        init(n * 2);
        int u, v;
        char c1, c2;
        while (m--)
        {
            scanf("%d%c%d%c", &u, &c1, &v, &c2);
            //i*2+1表示第i对的新郎,i*2表示第i对的新娘
            u *= 2;
            v *= 2;
            if (c1 == 'h') u++;
            if (c2 == 'h') v++;
            //添加限制条件，有关系的两个人可以同时在新娘那侧，但不能同时在新郎那侧
            addEdge(u, 1, v, 0);
            addEdge(v, 1, u, 0);
        }
        if (!check(n))
            printf("bad luck\n");
        else
        {
            n *= 2;
            //输出位于新娘那一侧的人
            for (int i = 2; i < n; i += 2)
            {
                if (vis[i << 1])
                    printf("%dw ", i / 2);
                else
                    printf("%dh ", i / 2);
            }
            printf("\n");
        }
    }
    return 0;
}