文章标题 HDU 5971 : Wrestling Match （2-sat）

Wrestling Match

题意：给出 n 个人，m 场 比赛 x 个已经确定的好人 y 个已经确定的坏人。每场比赛由 好人和坏人 组成。问是否能够将每个人划分成好人或者坏人。
分析：2-sat模板题，直接建图。注意：当有人既可以是好人又可以是坏人，应当输出no

代码：

#include<iostream>
#include<string>
#include<cstdio>
#include<cstring>
#include<vector>
#include<math.h>
#include<map>
#include<queue> 
#include<algorithm>

using namespace std;

const double pi=acos(-1.0);
const int inf = 0x3f3f3f3f;
const int maxn=4005;
const double eps=1e-8; 

vector <int> G[maxn];

int belong[maxn],stack[maxn],ins[maxn];
int sz[maxn],id,block,tp;
int dfn[maxn],low[maxn];

int n,m,x,y;

void init(){
    id=block=tp=0;
    for (int i=0;i<maxn;i++) G[i].clear();
    memset (dfn,0,sizeof (dfn));
    memset (low,0,sizeof (low));
    memset (ins,0,sizeof (ins));
}

void tarjan(int u){
    dfn[u]=low[u]=++id;
    ins[u]=1;//标记进栈 
    stack[++tp]=u;//进栈 
    for (int i=0;i<G[u].size();i++){
        int v=G[u][i];
        if (!dfn[v]){
            tarjan(v);
            low[u]=min(low[u],low[v]);
        }
        else if (ins[v]){
            low[u]=min(low[u],dfn[v]);
        }
    }
    if (low[u]==dfn[u]){
        block++;//强连通分量加一 
        int v;
        do{
            v=stack[tp--];
            ins[v]=0;
            belong[v]=block;//属于同一个连通块 
        }while (u!=v);
    }
}

int main()
{
    while (scanf ("%d%d%d%d",&n,&m,&x,&y)!=EOF){    
        int a,b;
        init();
        for (int i=0;i<m;i++){
            scanf ("%d%d",&a,&b);//一人为假一人为真 
            G[2*a].push_back(2*b+1);//a false , b true 
            G[2*b].push_back(2*a+1);//b false , a true
            G[2*a+1].push_back(2*b);//a true , b false
            G[2*b+1].push_back(2*a);//b true , a false
        }
        for (int i=0;i<x;i++){
            scanf ("%d",&a);//a是好人 
            G[2*a+1].push_back(2*a);
        } 
        for (int i=0;i<y;i++){
            scanf("%d",&a);//a是坏人 
            G[2*a].push_back(2*a+1);
        }
        for (int i=2;i<=2*n+1;i++){
            if (!dfn[i]){
                tarjan(i);
            }
        }
        bool ans=true;
        for (int i=1;i<=n;i++){
            if (belong[i*2]==belong[i*2+1]){
                ans=false;
                break;
            }
        }
        for (int i=1;i<=n;i++){
            if (G[i*2].empty()&&G[i*2+1].empty()){//既是好人也是坏人 
                ans=false;
                break;
            }
        }
        if (ans==false){
            printf ("NO\n");
        }
        else printf ("YES\n");
    }
    return 0;
 }