HDU1520Anniversary party(树形dp模板父子(相邻节点)不能同时选最多选多大)

最新推荐文章于 2022-06-01 20:06:48 发布

Helium_wild

最新推荐文章于 2022-06-01 20:06:48 发布

阅读量407

点赞数

CC 4.0 BY-SA版权

分类专栏： DP==树形dp

本文链接：https://blog.youkuaiyun.com/qq_42576687/article/details/98518504

DP==树形dp 专栏收录该内容

14 篇文章

订阅专栏

该博客介绍了如何解决一个树形动态规划问题，其中树的每个节点都有权值，但选取节点时禁止选择父子节点。目标是找出能获得的最大权值和，通过动态规划和深度优先搜索策略可以实现这一目标。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目
在这里插入图片描述 题意： 给你一颗树，每个点都有权值，有正有负，随意选取但是父子不能同时选取问你选取的最大的权值和是多少？

dp[x][0]：不选x节点 dp[x][1]：选x节点
dfs即可完成dp过程注意dp[x][1]+=val[x]一定刚进dfs就要这样。

dp[fa][0]+=max(dp[son][0],dp[son][1])
dp[fa][1]+=dp[son][0];

#include<bits/stdc++.h>
#define m(a,b) memset(a,b,sizeof a)
using namespace std;
const int N=6e3+5;
int val[N],dp[N][2];
struct Edge{int to,next;}edge[N<<1];
int head[N],tot;
void add(int from,int to){
    edge[++tot]=(Edge){to,head[from]},head[from]=tot;
    edge[++tot]=(Edge){from,head[to]},head[to]=tot;
}
void dfs(int x,int fa){
    dp[x][1]+=val[x];
    for(int i=head[x];i;i=edge[i].next){
        int y=edge[i].to;
        if(y==fa) continue;
        dfs(y,x);
        dp[x][0]+=max(dp[y][0],dp[y][1]);
        dp[x][1]+=dp[y][0];
    }
}
int main(){
    int n;
    while(~scanf("%d",&n)){
        m(dp,0),m(head,0),tot=0;
        for(int i=1;i<=n;++i) scanf("%d",&val[i]);
        int x,y,root=0;
        while(~scanf("%d%d",&x,&y),x||y) add(x,y);
        dfs(1,-1);
        printf("%d\n",max(dp[1][0],dp[1][1]));
    }
}