树形dp 的 一道巨简单题

我好菜

(这三个字写这么大完全是为了提醒自己 hhhh 竟然这都不会)
树形dp;
不懂得如何推出状态表达式 (即使再简单)看见题目就无从下手!!!有点烦

所以决定多刷一点dp题,
先看了最基础的:
洛谷P1352

这个题其实挺简单的,但是我是真的废。。。

题意:
给出一棵树,每个节点有自己的权值,选择出一些点,选择点的时候如果选择了当前节点的父节点则不能选择当前节点,这很显然的树形dp,可能是我做dp题目太少了吧,连dp数组都不知道该代表点啥。
于是看了题解。
dp[i][0] 代表这颗子树不选 i 节点 能选出的最大权值。
dp[i][1] 代表这颗子树上选择 i 节点能选出的最大权值。
于是得到转移方程:
dp[i][0] += max(dp[v][0],dp[v][1]);
dp[i][1] += dp[v][0]; dp[i][1] += val[i];
(这里v是 i 的子节点)
上代码:

#include<stdio.h>
#include<algorithm>
#include<vector>
using namespace std;
const int maxn = 6005;
int val[maxn];
int du[maxn];
vector<int> vv[maxn];
int dp[maxn][2];

void dfs(int x)
{
	dp[x][0] = 0;
	dp[x][1] = val[x];
	for (int i = 0; i < vv[x].size(); i ++ )
	{
		int v = vv[x][i];
		dfs(v);
		dp[x][0] += max(dp[v][0],dp[v][1]);
		dp[x][1] += dp[v][0];
	}
	
}

int main()
{
	int n;
	scanf("%d",&n);
	for (int i = 1; i <= n; i ++ )
	{
		scanf("%d",&val[i]);
	}
	for (int i = 1; i < n; i ++ )
	{
		int x,y;
		scanf("%d%d",&x,&y);
		du[x] ++ ;
		vv[y].push_back(x);
	}
	int r;
	for (int i = 1; i <= n; i ++ )
	{
		if(du[i] == 0)
		{
			r = i;
			break;
		}
	}
	dfs(r);
	printf("%d\n",max(dp[r][0],dp[r][1]));
	
}
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值