Wrestling Match【DFS+二分图染色】

题目链接
题意：
给你n, m, x, y四个数，n代表编号，接下来m行，每行两个数a,b，二者PK，其中一方为好人将不会输，接下来x个好人和y个坏人；问是否能够将每个人划分成好人或者坏人。
思路：
二分图染色问题，注意读题；
在已知的好人坏人中第一遍DFS，在余下的图中未知的位置，分别填上好人坏人的条件下看看是否都矛盾；

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cmath>
#define CLR(a, b) memset(a, b, sizeof(a))
using namespace std;
typedef long long LL;

const int MAXN = 2e5 + 10;

int head[MAXN], res[MAXN], sta[MAXN];
int cnt = 0;
int n, m, X, Y;

struct node {
    int to;
    int next;
}edge[MAXN << 1];

void add_edge(int u, int v) {
    edge[cnt].to = v;
    edge[cnt].next = head[u];
    head[u] = cnt++;
}

bool dfs(int u, int s) {
    //printf("# : %d %d\n", u, s);
    if(sta[u] != -1 && sta[u] != s) return false;
    if(sta[u] == s) return true;
    sta[u] = s;
    for(int i = head[u]; i != -1; i = edge[i].next) {
        int v = edge[i].to;
        if(sta[v] == -1) dfs(v, 1 - s);
        if(sta[v] != -1 && !dfs(v, 1- s)) return false;
        //if(v != u) if(!dfs(v, 1 - s)) return false;
    }
    return true;
}

int main() {
    while(scanf("%d %d %d %d", &n, &m, &X, &Y) != EOF) {
        bool flag = true;
        CLR(head, -1);
        CLR(sta, -1);
        CLR(res, 0);cnt = 0;
        for(int i = 0; i < m; i++) {
            int u, v;
            scanf("%d %d", &u, &v);
            add_edge(u, v);
            add_edge(v, u);
        }
        int k;
        for(int i = 0; i < X + Y; i++) {
            scanf("%d", &k);
            res[k] = 1;
            if(!dfs(k, i < X ? 1 : 0)) flag = false;
        }
        for(int i = 1; i <= n; i++) {
            if(sta[i] == -1) {
               // printf("*%d\n", head[i]);
                if(head[i] == -1 || (!dfs(i, 0) && !dfs(i, 1))){
                    flag = false;
                    break;
                }
            }
        }
        puts(flag ? "YES" : "NO");
    }
    return 0;
}