Wrestling Match HDU - 5971(二分图染色)

Wrestling Match HDU - 5971

题意：

有n个人，问是否能分成好人和坏人。（对于只有一个点的联通块无法确定，除非事先知道他是好人还是坏人）

思路：

就是二分图染色（dfs）

1. 先对好人dfs（假如只有一个点的联通块，那么就跳过）
2. 之后对坏人dfs（假如只有一个点的联通块，那么就跳过）
3. 最后对所有人dfs（假如只有一个点的联通块，无法判断，输出no）

AC

#include <iostream>
#include <cstdio>
#include <cstring>
#include <vector>
#define mst(x,a) memset(x,a,sizeof(x))
#define For(i,x,y) for(int i=(x); i<=(y); i++)
#define pb push_back
using namespace std;
const int maxn=1e3+10;
vector<int>e[maxn];
int n,m,x,y;
int color[maxn];
vector<int>vx,vy;
bool dfs(int u){
    if(e[u].size()==0)return false;
    for(int i=0; i<e[u].size(); i++){
        int v=e[u][i];
        if(color[u]==color[v])return false;
        if(!color[v]){
            color[v]=3-color[u];
            if(!dfs(v))return false;
        }
    }
    return true;
}
int fl;
void solve1(int x, vector<int>v){
    For(i,0,x-1){
        int id=v[i];
        if(e[id].size()==0)continue;
        if(dfs(id))continue;
        fl=0;
        break;
    }
}
void ans(int a){
    if(a)cout<<"YES"<<endl;
    else cout<<"NO"<<endl;
}
int main()
{
    ios::sync_with_stdio(0);
    while(cin>>n>>m>>x>>y){
        int u,v;
        fl=1;
        vx.clear();vy.clear();
        For(i,1,n)e[i].clear();
        For(i,1,m)cin>>u>>v,e[u].pb(v),e[v].pb(u);
        mst(color,0);
        For(i,1,x)cin>>u,vx.pb(u),color[u]=1;
        For(i,1,y)cin>>u,vy.pb(u),color[u]=2;
        solve1(x,vx);
        solve1(y,vy);
        For(id,1,n){
            if(!color[id]){
                color[id]=1;
                if(dfs(id))continue;
                fl=0;
                break;
            }
        }
        ans(fl);
    }
    return 0;
}