codeforces round 627 div3 [树上dp]F. Maximum White Subtree

IZUMIXINGI

于 2020-10-15 10:31:02 发布

阅读量130

点赞数

分类专栏： dp dfs

本文链接：https://blog.youkuaiyun.com/qq_20252251/article/details/109092284

版权

dp 同时被 2 个专栏收录

20 篇文章

订阅专栏

dfs

1 篇文章

订阅专栏

本文展示了一段C++代码，用于解决树形结构中的问题。通过深度优先搜索（DFS）和动态规划（DP）计算每个节点的最优点，然后使用DFS进行传递更新。该算法在N个节点的图上运行，处理邻接表并优化了路径的计算。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

在这里插入图片描述

#include<iostream>
#include<vector>
#define pb push_back
 
using namespace std;
const int N = 2e5 + 5;
vector<int> g[N];
int dp[N],a[N];
const int INF = 0x3f3f3f3f;
int dfs(int u,int pre){
	int ans = (a[u] == 1)?1 : -1;	
	for(unsigned int i = 0;i < g[u].size();i++){
		int v = g[u][i];
		if(v != pre)ans += dfs(v,u);
	}
	dp[u] = ans;
	return max(ans,0);
}
void dfs1(int u,int pre){
	for(unsigned int i = 0;i < g[u].size();i++){
		int v = g[u][i];
		if(v != pre && dp[u] - max(dp[v],0) > 0){
			dp[v] = dp[u] - max(dp[v],0) + dp[v];
		}
		if(v != pre)dfs1(v,u);
	}
}

int main(){
	int n;
	cin>>n;
	for(int i = 1;i <= n;i++)cin>>a[i],dp[i] = -INF;
	for(int i = 1;i <= n - 1;i++){
		int u,v;
		cin>>u>>v;
		g[u].pb(v);
		g[v].pb(u);
	}
	dfs(1,-1);
	dfs1(1,-1); 
	for(int i = 1;i <= n;i++)
		cout<<dp[i]<<" ";
	return 0;
}