最大子树和
树形dp
对于一个节点来说,如果删除掉一个连接子节点的边,则以该子节点为根的子树上面的贡献都会变成 \(0\)
设计状态:\(dp[u]\),表示以 \(u\) 为根的子树中,贡献值最大为多少
状态转移:\(dp[u] = max(0, dp[v])\),\(v\) 为 \(u\) 的子节点,\(0\) 的话表示删除该边,\(dp[v]\) 的话表示保留该边
注意:最后一定要至少要存在一个节点
答案不一定以 \(1\) 为根,所有子树都有可能是答案
#include <iostream>
#include <cstdio>
#include <algorithm>
#include <vector>
#include <array>
using namespace std;
typedef long long ll;
vector<ll>dp;
vector<vector<int>>gra;
void dps(int now, int pre)
{
for(int nex : gra[now])
{
if(nex == pre) continue;
dps(nex, now);
dp[now] += max(0ll, dp[nex]);
}
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(0);
cout.tie(0);
int n;
cin >> n;
dp.resize(n + 1);
gra.resize(n + 1);
for(int i=1; i<=n; i++) cin >> dp[i];
for(int i=1; i<n; i++)
{
int a, b;
cin >> a >> b;
gra[a].push_back(b);
gra[b].push_back(a);
}
dps(1, 1);
ll ans = dp[1];
for(int i=1; i<=n; i++) ans = max(ans, dp[i]);
cout << ans << endl;
return 0;
}
原文地址:http://www.cnblogs.com/dgsvygd/p/16849017.html
1. 本站所有资源来源于用户上传和网络,如有侵权请邮件联系站长!
2. 分享目的仅供大家学习和交流,请务用于商业用途!
3. 如果你也有好源码或者教程,可以到用户中心发布,分享有积分奖励和额外收入!
4. 本站提供的源码、模板、插件等等其他资源,都不包含技术服务请大家谅解!
5. 如有链接无法下载、失效或广告,请联系管理员处理!
6. 本站资源售价只是赞助,收取费用仅维持本站的日常运营所需!
7. 如遇到加密压缩包,默认解压密码为"gltf",如遇到无法解压的请联系管理员!
8. 因为资源和程序源码均为可复制品,所以不支持任何理由的退款兑现,请斟酌后支付下载
声明:如果标题没有注明"已测试"或者"测试可用"等字样的资源源码均未经过站长测试.特别注意没有标注的源码不保证任何可用性