练习题 No.22 二分图判定

要求

给定一个具体n个顶点的图。要给图上每个顶点染色，并且要使相邻的顶点的颜色不同。问是否能最多用2种颜色进行染色？题目保证没有重边和自环。

限制条件

• 1 <= n <= 1000

输入格式

第一行输入 n和e(边数)
接下来e行
每行输入开始边和结束边

输出格式

输出YES和NO

测试输入

4 4
0 3
0 1
1 2
2 3

测试输出

YES

解题思路

深度搜索每个顶点，并且进行交替染色即可。

代码

#include <iostream>
#include <vector>

using namespace std;  

class Vertex {
    public:
        vector<Vertex *> edge; 
        int color;
        Vertex() {
            color = 0;
        }
}; 

int V, E;
Vertex g[101];

bool dfs(Vertex &v, int c) {
    v.color = c;
    for (int i = 0; i < v.edge.size(); i++) {
        if (v.edge[i]->color == c) {
            return false;
        } else if (v.edge[i]->color == 0 && !dfs(*v.edge[i], -c)) {
            return false;
        }
    }
    return true;
}

int main() {
    cin >> V >> E;
    for (int i = 0; i < E; i++) {
        int s, t;
        cin >> s >> t;
        g[s].edge.push_back(&g[t]);
    }
    for (int i = 0; i < V; i++) {
        if (g[i].color == 0) {
            if (!dfs(g[i], 1)) {
                cout << "No" << endl;
                return 0;
            }
        }
    }
    cout << "YES" << endl;
    return 0;  
}