打卡信奥刷题（3486）用C++实现信奥题 P10723 [GESP202406 七级] 黑白翻转

原创于 2026-08-03 07:14:57 发布 · 124 阅读

3 ·

本内容遵循CC 4.0 BY-SA版权协议

GEO检测

收录于

C++

P10723 [GESP202406 七级] 黑白翻转

题目背景

对应的选择、判断题：https://ti.luogu.com.cn/problemset/1155

题目描述

小杨有一棵包含 $n$ 个节点的树，这棵树上的任意一个节点要么是白色，要么是黑色。小杨认为一棵树是美丽树当且仅当在删除所有白色节点之后，剩余节点仍然组成一棵树。

小杨每次操作可以选择一个白色节点将它的颜色变为黑色，他想知道自己最少要执行多少次操作可以使得这棵树变为美丽树。

输入格式

第一行包含一个正整数 $n$ ，代表树的节点数。

第二行包含 $n$ 个非负整数 $a_1,a_2,\ldots,a_n$ ，其中如果 $a_i=0$ ，则节点 $i$ 的颜色为白色，否则为黑色。

之后 $n - 1$ 行，每行包含两个正整数 $x_i,y_i$ ，代表存在一条连接节点 $x_i$ 和 $y_i$ 的边。

输出格式

输出一个整数，代表最少执行的操作次数。

输入输出样例 #1

输入 #1

输出 #1

说明/提示

样例解释

将节点 $1$ 和 $3$ 变为黑色即可使这棵树变为美丽树，此时删除白色节点 $5$ ，剩余黑色节点仍然组成一棵树。

数据范围

子任务编号	数据点占比	$n$	$a_i$	特殊条件
$1$	$30\%$	$\leq 10^5$	$0\leq a_i\leq 1$	树的形态为一条链
$2$	$30\%$	$\leq 10^5$	$0\leq a_i\leq 1$	只有两个节点颜色为黑色
$3$	$40\%$	$\leq 10^5$	$0\leq a_i\leq 1$

对于全部数据，保证有 $1\leq n\leq 10^5$ ， $0\leq a_i\leq 1$ 。

C++实现

#include <bits/stdc++.h>
using namespace std;

bool color[100005];
int ans, n;
vector <int> e[100005];

bool dfs(int x, int fa) {
	bool t = 0;
	for (int i = 0; i < e[x].size(); i++) {
		int to = e[x][i];
		if (to != fa && dfs(to, x)) t = 1;  
	}
	if (!color[x] && t) ans++;
	return (color[x] || t); 
}

int main() {
	cin >> n;
	for (int i = 1; i <= n; i++) cin >> color[i];
	for (int i = 1; i < n; i++) {
		int x, y;
		cin >> x >> y;
		e[x].push_back(y); e[y].push_back(x);
	}
	int x = -1;
	for (int i = 1; i <= n; i++)
		if (color[i]) { // 找到黑色点 
			x = i; break;
		}
	dfs(x, 0);
	cout << ans;
}