hdu4115Eliminate the Conflict

链接：http://acm.hdu.edu.cn/showproblem.php?pid=4115

题意：两个人猜拳，B提前知道了A的出手情况，但是对B的出手有要求，(a,b,k)当k=0时要求B的第a次出手和第b次出手要一样，当k=1时要求B的第a次出手和第b次出手不一样。问B能否一局都不输。

分析：将B的每次出手分解成3种（石头，剪刀，布）情况，然后根据题目要求对其进行真假限制建立2-sat模型就行了。

代码：

#include<map>
#include<set>
#include<stack>
#include<cmath>
#include<queue>
#include<bitset>
#include<math.h>
#include<vector>
#include<string>
#include<stdio.h>
#include<cstring>
#include<iostream>
#include<algorithm>
#pragma comment(linker, "/STACK:102400000,102400000")
using namespace std;
typedef double db;
typedef long long ll;
typedef unsigned int uint;
typedef unsigned long long ull;
const db eps=1e-5;
const int N=6e4+10;
const int M=2e6+10;
const ll MOD=1000000007;
const int mod=1000000007;
const int MAX=1000000010;
const double pi=acos(-1.0);
struct twosat {
    int n,k,d[N];
    bool mark[N];
    int tot,u[N],v[M],pre[M];
    void init(int a) {
        n=a<<1;tot=0;
        for (int i=0;i<n;i++) u[i]=-1,mark[i]=false;
    }
    void add_Edge(int a,int aval,int b,int bval) {
        a=a*2+aval;b=b*2+bval;
        v[tot]=b;pre[tot]=u[a^1];u[a^1]=tot++;
        v[tot]=a;pre[tot]=u[b^1];u[b^1]=tot++;
    }
    bool dfs(int a) {
        if (mark[a]) return true;
        if (mark[a^1]) return false;
        mark[a]=true;d[k++]=a;
        for (int i=u[a];~i;i=pre[i])
        if (!dfs(v[i])) return false;
        return true;
    }
    bool solve() {
        int i,j;k=0;
        for (i=0;i<n;i+=2)
        if (!mark[i]&&!mark[i^1]) {
            if (!dfs(i)) {
                for (j=0;j<k;j++) mark[d[j]]=false;
                k=0;if (!dfs(i^1)) return false;
            }
        }
        return true;
    }
}TS;
int a[N];
int main()
{
    int i,n,m,t,ca,A,B,K;
    scanf("%d", &t);
    for (ca=1;ca<=t;ca++) {
        scanf("%d%d", &n, &m);
        TS.init(3*n);
        for (i=1;i<=n;i++) {
            scanf("%d", &a[i]);
            A=3*(i-1);
            if (a[i]==1) {
                TS.mark[(A+2)<<1|1]=true;
                TS.add_Edge(A,0,A+1,0);
                TS.add_Edge(A,1,A+1,1);
            } else if (a[i]==2) {
                TS.mark[A<<1|1]=true;
                TS.add_Edge(A+1,0,A+2,0);
                TS.add_Edge(A+1,1,A+2,1);
            } else {
                TS.mark[(A+1)<<1|1]=true;
                TS.add_Edge(A,0,A+2,0);
                TS.add_Edge(A,1,A+2,1);
            }
        }
        for (i=1;i<=m;i++) {
            scanf("%d%d%d", &A, &B, &K);
            A=3*(A-1);B=3*(B-1);
            if (K==0) {
                TS.add_Edge(A,0,B,1);
                TS.add_Edge(A+1,0,B+1,1);
                TS.add_Edge(A+2,0,B+2,1);
            } else {
                TS.add_Edge(A,1,B,1);
                TS.add_Edge(A+1,1,B+1,1);
                TS.add_Edge(A+2,1,B+2,1);
            }
        }
        if (TS.solve()) printf("Case #%d: yes\n", ca);
        else printf("Case #%d: no\n", ca);
    }
    return 0;
}