树形dp模板
入门题目:
#include<bits/stdc++.h>
using namespace std;
int f[16001];int cnt[16001];int a[16001];
vector<int>e[16001];int n;int mx=-10000000;
void dfs(int n,int fa)
{
f[n]=a[n];
for(int i=0;i<e[n].size();i++)
{
if(e[n][i]!=fa)
{
dfs(e[n][i],n);
if(f[e[n][i]]>0)
f[n]+=f[e[n][i]];
}
}
}
int main()
{
cin>>n;
for(int i=1;i<=n;i++)
cin>>a[i];
for(int i=1;i<n;i++)
{
int x,y;
cin>>x>>y;
e[x].push_back(y);
e[y].push_back(x);
}
dfs(1,0);
for(int i=1;i<=n;i++)
{
if(f[i]>mx)mx=f[i];
}
cout<<mx<<endl;
}
简单的dfs遍历,相当于线性dp中的1~n
dp方程设计在dfs的for循环中
本文作者:邓佑孤
本文链接:https://www.cnblogs.com/Arc-ux/p/18563701
版权声明:本作品采用知识共享署名-非商业性使用-禁止演绎 2.5 中国大陆许可协议进行许可。
【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步