我好菜
(这三个字写这么大完全是为了提醒自己 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]));
}