CF1626E Black and White Tree (换根 dp)
换根 dp
树上路径行走问题,因其节点的转移不止于其子树有关,一般考虑换根 dp 或寻找新的转移顺序。
在这题里,考虑一个以
-
-
-
且
对于第三种方式,此时
那么这样的 dp 形式直接用换根 dp 即可。先求出以
#include <bits/stdc++.h>
#define pii std::pair<int, int>
#define fi first
#define se second
#define pb push_back
typedef long long i64;
const i64 iinf = 0x3f3f3f3f, linf = 0x3f3f3f3f3f3f3f3f;
const int N = 3e5 + 10;
int n;
std::vector<int> e[N];
int a[N], sz[N], f[N], g[N];
void dfs(int u, int fa) {
sz[u] = a[u];
if(a[u]) f[u] = 1;
for(auto v : e[u]) {
if(v == fa) continue;
if(a[v]) f[u] = 1;
dfs(v, u);
if(f[v] && sz[v] > 1) f[u] = 1;
sz[u] += sz[v];
}
}
void dfs2(int u, int fa) {
if(f[u] || a[u]) g[u] = 1;
for(auto v : e[u]) {
if(v == fa) continue;
if(a[u]) g[v] = 1;
if(g[u] && sz[1] - sz[v] > 1) g[v] = 1;
dfs2(v, u);
}
}
void Solve() {
std::cin >> n;
for(int i = 1; i <= n; i++) std::cin >> a[i];
for(int i = 1; i < n; i++) {
int u, v;
std::cin >> u >> v;
e[u].pb(v), e[v].pb(u);
}
dfs(1, 0);
dfs2(1, 0);
for(int i = 1; i <= n; i++) {
std::cout << g[i] << " \n"[i == n];
}
}
int main() {
std::ios::sync_with_stdio(false);
std::cin.tie(nullptr);
Solve();
return 0;
}
标签:
DP
Buy me a cup of coffee ☕.
【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· 阿里最新开源QwQ-32B,效果媲美deepseek-r1满血版,部署成本又又又降低了!
· 开源Multi-agent AI智能体框架aevatar.ai,欢迎大家贡献代码
· Manus重磅发布:全球首款通用AI代理技术深度解析与实战指南
· 被坑几百块钱后,我竟然真的恢复了删除的微信聊天记录!
· AI技术革命,工作效率10个最佳AI工具