文章標題 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;
 }