换根DP CF1324F Maximum White Subtree

最新推荐文章于 2024-09-15 22:44:29 发布

weixin_49373560

最新推荐文章于 2024-09-15 22:44:29 发布

阅读量83

点赞数

分类专栏：树形DP 文章标签：深度优先图论算法

本文链接：https://blog.csdn.net/weixin_49373560/article/details/126875738

版权

树形DP 专栏收录该内容

2 篇文章 0 订阅

订阅专栏

这篇文章讲解了解决CF1324F Maximum White Subtree问题的动态规划方法，通过自底向上和自顶向下的转移策略，寻找含有最多白色节点且黑白点差值最大的连通子图。博主详细介绍了转移过程和代码实现，适合理解并应用动态规划解决图论问题的读者。

摘要由CSDN通过智能技术生成

CF1324F Maximum White Subtree

Description

给定一棵 $n$ 个节点无根树，节点 $u$ 颜色为 $a_u$ ，若 $a_u=0$ 为黑点， $a_u=1$ 为黑点。
对于每个节点 $u$ ，选出一个包含 $u$ 的连通子图，设子图白点个数为 $cnt_1$ ，黑点个数为 $cnt_2$ ，求最大化 $cnt_1 - cnt_2$ 。
$\leq n \leq 2 \times 10^5$ ， $\leq a_u \leq 1$ 。

Solution

下午水了一道绿题。。。
我们设 $r t = 1$ ，这样考虑一个根节点的最优。
显然自下向上转移即可：
设 $f [i]$ 为以 $i$ 为子树的最大值，则对 $u ， v$ 有：
$if(f[v]>=0)\quad f[u]+=f[v];$

然后考虑由上至下转移：
无论怎样，u和fa可在同一个最优联通子图中或（断掉向上路径）和子树的一部分成图。
1.若 $u$ 对 $f a$ 产生贡献（f[u]>=0），则： $an s [u] = min (an s [v], f [u]);$
2.若 $u$ 对 $f a$ 无贡献（f[u]<0），则： $an s [u] = min (an s [v], - 1 ll);$
转移代码如下：

	if(u==rt) ;//rt根节点的ans 先前第一遍dfs求出
	else if(f[u]>=0) ans[u]=max(ans[fa], f[u]);
	else ans[u]=max(ans[fa]-1, -1ll);

Code

ll f[maxn], ans[maxn];
void dfs(ll u, ll fa){
	f[u]=a[u]?1:-1;
	for(ll i = hd[u]; i ; i=e[i].nxt){
		ll v=e[i].to;
		if(v==fa) continue;
		dfs(v, u);
		
		if(f[v]>0) f[u]+=f[v];//产生（向上）贡献
	}
}
void down(ll u, ll fa){
	if(u==rt) ;//根节点
	else if(f[u]>=0) ans[u]=max(f[u], ans[fa]);//f[u]>=0:有贡献
	else ans[u]=max(ans[fa]-1, -1ll);//f[u]<0:无贡献
	for(ll i = hd[u]; i ; i=e[i].nxt){
		ll v=e[i].to;
		if(v==fa) continue;
		down(v, u);//向下转移
	}
}

int main(){
//	freopen(".in", "r", stdin);
//	freopen(".out", "w", stdout);
	//...初始化...
	//...建边...
	rt=1;
	dfs(rt, 0);
	ans[rt]=f[rt];//根节点答案
	down(rt, 0);
	
	//print：ans
	return 0;
}