题目描述: 给定一棵有根树,每条边有一种颜色,每个节点的颜色可能为黑色或白色。现在要对该树进行染色,使得每个节点都被染色,且相邻节点的颜色不能相同,问是否能完成染色任务。
思路:
代码示例:
#include
#include
using namespace std;
const int MAXN = 100010;
vector graph[MAXN];
int color[MAXN]; // 每个节点的颜色,0表示未染色,1表示黑色,2表示白色
bool dfs(int cur, int pre, int c) {
color[cur] = c;
for (int i = 0; i < graph[cur].size(); i++) {
int next = graph[cur][i];
if (next == pre) continue; // 避免重复访问
if (color[next] == 0 && !dfs(next, cur, 3 - c)) { // 递归向下,取相反的颜色
return false; // 如果下面没法染色,则返回无法染色
} else if (color[next] == c) { // 和父节点颜色相同,无法染色
return false;
}
}
return true;
}
int main() {
int n, m;
cin >> n >> m;
for (int i = 0; i < m; i++) {
int u, v;
cin >> u >> v;
graph[u].push_back(v);
graph[v].push_back(u); // 双向边
}
if (dfs(1, 0, 1)) { // 从第一个节点