2016年ICPC亚洲区域赛大连站A题Wrestling Match

题目链接https://ac.nowcoder.com/acm/problem/124640

解题思路：深搜 + 二分图染色问题

#include <bits/stdc++.h>
using namespace std;
vector<int> mp[1010]; //mp[i]存储i这个选手的所有出边 
int n; //总选手数 
int color[1010]; //存储每个选手是good还是bad 
bool dfs(int p, int c) { //深搜，假设p选手是c的话，看是否有矛盾出现
    //有矛盾返回false,无矛盾返回true 
    color[p] = c;
    for (int i = 0 ; i < mp[p].size(); i++) { //遍历p的所有出边 
        if (color[mp[p][i]] == c) {
            return false;
        }
        if (color[mp[p][i]] == 0 && !dfs(mp[p][i], -c)) { //如果该点还没被染色，就把它染为另外一种颜色 
            return false;
        }
    }
    return true;
}
void solve() {
    for (int i = 1; i <= n; i++) {  
        if (color[i] == 1) { //如果该选手是good的话 
            if (!dfs(i, 1)) {
                cout << "NO" << endl;
                return;
            }
        }
    }
    for (int i = 1; i <= n; i++) {
        if (color[i] == -1) { //如果该选手是bad的话 
            if (!dfs(i, -1)) {
                cout << "NO" << endl;
                return;
            }
        }
    }
    for (int i = 1; i <= n; i++) {
        if (color[i] == 0) {
            if (!dfs(i, 1)) {
                cout << "NO" << endl;
                return;
            }
        }
    }
    for (int i = 1; i <= n; i++) {
        if (color[i] == -5) {
            cout << "NO" << endl;
            return;
        }
    }
    cout << "YES" << endl;
}
int main() {
    int m, x, y; //总边数，good选手数，bad选手数 
    while (cin >> n >> m >> x >> y) {
        for (int i = 1; i <= n; i++) { //每一次都要把i这个选手的所有出边清空并赋初值-5 
            mp[i].clear();
            color[i] = -5; //-5表示i选手的是good还是bad是未知的 
        }
        for (int i = 0; i < m; i++) { //遍历m场比赛 
            int a, b;
            cin >> a >> b; //a选手和b选手之间有场比赛 
            color[a] = 0; //0表明该选手有比赛，需要被染色 
            color[b] = 0; //0表明该选手有比赛，需要被染色 
            mp[a].push_back(b); //a连向b有一条边 
            mp[b].push_back(a); //b连向a有一条边 
        }
        for (int i = 0; i < x; i++) { //x个已知的good选手 
            int t;
            cin >> t;
            color[t] = 1; //1表示该选手是good选手 
        }
        for (int i = 0; i < y; ++i) { //y个已知的bad选手 
            int t;
            cin >> t;
            color[t] = -1; //-1表示该选手是bad 
        }
        solve(); //解决问题 
    }
    return 0;
}