Day 1 树状dp
树状dp的水题(放假第一天摸个鱼QAQ):题目很明显是个树状dp了,所以开一个dp数组存储每一个点作为根节点时的最优解,循环一遍目标节点的子节点的dp值,为正则有留下来的价值,目标节点加上其dp值即可,最后跑一遍循环找出最优解。
代码如下(示例):
#include <bits/stdc++.h>
#define int long long
using namespace std;
int N,f,e;
int A[16005];
vector<int>Son[16005];
int Dp[16005];
int Final=-0x7ffffff;
void Dfs(int now,int dad){
int k;
Dp[now]=A[now];
for(int i=0;i<Son[now].size();i++){
k=Son[now][i];
if(k!=dad){
Dfs(k,now);
if(Dp[k]>0)
Dp[now]+=Dp[k];
}
}
return ;
}
signed main(){
cin>>N;
for(int i=1;i<=N;i++)
cin>>A[i];//美丽指数
for(int i=1;i<=N-1;i++){
cin>>f>>e;
Son[f].push_back(e);
Son[e].push_back(f);//互相作为自己的儿子
}
Dfs(1,0);//Dfs(son,father)自上向下搜索
for(int i=1;i<=N;i++)
Final=max(Final,Dp[i]);
cout<<Final<<endl;
return 0;
}