树形dp
树形dp
还有一天半
不知道学什么好了
心情起伏波动极大
没有上司的舞会
- 题意:给定一棵树,每个点有快乐值,父亲和儿子不能同时存在,求最大快乐值。
首先分析父节点和子节点的关系
表示结点i不选,表示选
父节点影响自己点
对于父节点x的每条边i:
所以我们递归到叶子结点,回溯+dp即可
#include<iostream>
#include<cstdio>
#include<vector>
using namespace std;
vector<int> a[6005];
int n, dp[6005][2], v[6005], w[6005], root ;
void dfs(int x){
dp[x][0] = 0;
dp[x][1] = w[x];
for(int i = 0; i < a[x].size(); i++){
int y = a[x][i];
dfs(y);
dp[x][0] += max(dp[y][1], dp[y][0]);
dp[x][1] += dp[y][0];
}
}
int main(){
cin >> n;
for(int i = 1; i <= n; i++) cin >> w[i];
for(int i = 1; i < n; i++){
int x, y;
cin >> x >> y;
v[x] = 1; //x has a father
a[y].push_back(x); // x is a son of y
}
// find the root
for(int i = 1; i <= n; i++){
if(!v[i]) root = i;
}
dfs(root);
cout << max(dp[root][0], dp[root][1]);
return 0;
}
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】凌霞软件回馈社区,博客园 & 1Panel & Halo 联合会员上线
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· 深入理解 Mybatis 分库分表执行原理
· 如何打造一个高并发系统?
· .NET Core GC压缩(compact_phase)底层原理浅谈
· 现代计算机视觉入门之:什么是图片特征编码
· .NET 9 new features-C#13新的锁类型和语义
· Spring AI + Ollama 实现 deepseek-r1 的API服务和调用
· 《HelloGitHub》第 106 期
· 数据库服务器 SQL Server 版本升级公告
· 深入理解Mybatis分库分表执行原理
· 使用 Dify + LLM 构建精确任务处理应用