[POJ3678]Katu Puzzle（2-SAT）

题目描述

传送门

题解

很显然的2-SAT问题
需要注意的一点是，如果一个必选1或必选0，那么就从不必选的向必选的连边

代码

#include<algorithm>
#include<iostream>
#include<cstring>
#include<cstdio>
#include<cmath>
using namespace std;
#define N 4000005

char opt[10];
int n,m,a,b,c,dfs_clock,scc,top;
int tot,point[N],nxt[N],v[N];
int dfn[N],low[N],belong[N],stack[N],vis[N];
bool flag;

void add(int x,int y)
{
    ++tot; nxt[tot]=point[x]; point[x]=tot; v[tot]=y;
}
void tarjan(int x)
{
    dfn[x]=low[x]=++dfs_clock;stack[++top]=x;vis[x]=1;
    for (int i=point[x];i;i=nxt[i])
        if (!dfn[v[i]])
        {
            tarjan(v[i]);
            low[x]=min(low[x],low[v[i]]);
        }
        else if (vis[v[i]])
            low[x]=min(low[x],dfn[v[i]]);
    if (dfn[x]==low[x])
    {
        int now=0;++scc;
        while (now!=x)
        {
            now=stack[top--];
            belong[now]=scc;
            vis[now]=0;
        }
    }
}
int main()
{
    scanf("%d%d",&n,&m);
    for (int i=1;i<=m;++i)
    {
        scanf("%d%d%d",&a,&b,&c);
        scanf("%s",opt);
        ++a,++b;
        if (opt[0]=='A')
        {
            if (!c) add(a+m,b),add(b+m,a);
            else add(a,a+m),add(b,b+m);
        }
        else if (opt[0]=='O')
        {
            if (!c) add(a+m,a),add(b+m,b);
            else add(a,b+m),add(b,a+m);
        }
        else if (opt[0]=='X')
        {
            if (!c) add(a,b),add(b,a),add(a+m,b+m),add(b+m,a+m);
            else add(a,b+m),add(b,a+m),add(a+m,b),add(b+m,a);
        }
    }
    for (int i=1;i<=2*m;++i)
        if (!dfn[i]) tarjan(i);
    flag=true;
    for (int i=1;i<=m;++i)
        if (belong[i]==belong[i+m])
            {flag=false;break;}
    if (flag) puts("YES");
    else puts("NO");
}