codeforces1324F Maximum White Subtree_subtree codeforces-优快云博客

本文链接：https://blog.youkuaiyun.com/liufengwei1/article/details/104841199

本文介绍了一个使用树形动态规划方法解决特定子图最大值问题的算法实现。通过两次DFS遍历，分别计算向下子图和向上子图所能获得的最大值，最终求得每个节点作为根节点时子图的最大值。

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

https://codeforces.com/problemset/problem/1324/F

理解样例20分钟，写题10分钟系列

subtree不是子树是子图

树形DP维护sonval[u]表示以1位根，向下的子图能得到的最大值

然后第二遍维护faval，向上的子图能得到的最大值。

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

const int maxl=2e5+10;

int n;
int a[maxl],ans[maxl];
int sonval[maxl];
bool vis[maxl];
vector<int> e[maxl];

inline void dfs1(int u,int fa)
{
	if(a[u]==1)
		sonval[u]=1;
	else
		sonval[u]=-1;
	for(int v:e[u])
	{
		if(v==fa) continue;
		dfs1(v,u);
		if(sonval[v]>0)
			sonval[u]+=sonval[v];
	}
}

inline void prework()
{
	scanf("%d",&n);
	for(int i=1;i<=n;i++)
		scanf("%d",&a[i]);
	int u,v;
	for(int i=1;i<n;i++)
	{
		scanf("%d%d",&u,&v);
		e[u].push_back(v);
		e[v].push_back(u);
	}
	dfs1(1,0);
}

inline void dfs2(int u,int fa,int faval)
{
	ans[u]=faval+sonval[u];
	for(int v:e[u])
	{
		if(v==fa) continue;
		if(sonval[v]>0)
			dfs2(v,u,max(ans[u]-sonval[v],0));
		else
			dfs2(v,u,max(ans[u],0));
	}
}

inline void mainwork()
{
	dfs2(1,0,0);
}

inline void print()
{
	for(int i=1;i<=n;i++)
		printf("%d%c",ans[i],(i==n)?'\n':' ');
}

int main()
{
	prework();
	mainwork();
	print();
	return 0;
}