今天我们来简单了解一下树形dp,树形dp,顾名思义就是在树上进行动态规划,其实这和其他的动态规划中心思想是一样的,不过这个难点之处就是在树上进行的,因为树有其特殊的结构特点,一般对树的处理用的都是递归的思想,这就是其难点所在吧,其实学树形dp,HIA有助于帮助我们增强我们对树的处理,对于树形dp感觉也没有什么可以 说的,还是直接给题去训练吧,在做题中慢慢的理解树形dp的精髓。
题意:在一个公司中,每个人都有一个自己的直接上司(老板肯定是没有自己的上司的,)每个人都有自己的快乐值,现在要开一个paty,假如一个员工的直接上司去了,那么这个员工肯定是不会去了,现在让你挑选一部分人去参加paty,使参加的Paty的人的快乐值总和最大,让你输出这个最大的快乐值,
这个题一看就是一个树形dp的入门题,首先我们先把这个树形抛弃,只看dp,那么是不是很好理解呢,我们先定义一个数组, dp[maxn][2],dp[root][1] 代表root这个人去,dp[root][0] 代表这个人不去,如果这个人去那么他的下属是肯定去不了的,如果这个人不去,那么这个人的下属是可去可不去的,这样我们就可以得出动态转移方程, dp[r][1]+=dp[i][0]; dp[r][0]+=max(dp[i][0],dp[i][1]);到这里这个题是不是已经解决了一大半了,剩下的就是我们应该如何建立一个树,并去处理 他,
怎么建一个树呢,其实很简单,只要我们开一个数组记录每个人的上司就行了,然后再递归查询,这个树就出来了,
唉。感觉越说越乱,还是直接给代码吧,用代码再详细的讲解一下。
#include<stdio.h>
#include<string.h>
#include<iostream>
using namespace std;
const int maxn=6004;
int dp[maxn][2];//dp[i][0]代表i这个点不取,dp[i][1]代表这个点取。
bool vis[maxn];//标记每个点,防止重复.
int father[maxn];//记录每个子节点的父亲节点
int n,root;
void void_tree(int r) //这里运用了递归,请读者好好的理解一下,
{
vis[r]=1;
for(int i=1;i<=n;i++)
{
if(!vis[i]&&father[i]==r)
{
void_tree(i);
dp[r][1]+=dp[i][0];
dp[r][0]+=max(dp[i][0],dp[i][1]);
}
}
}
int main()
{
memset(dp,0,sizeof(dp));
memset(vis,0,sizeof(vis));
memset(father,0,sizeof(father));
root=0;
scanf("%d",&n);
for(int i=1;i<=n;i++)
{
scanf("%d",&dp[i][1]);
}
int s,t;
while(scanf("%d %d",&s,&t)&&(s||t))
{
father[s]=t;
}
/* for(int i=1;i<=n;i++)
printf("%d %d\n",i,father[i]);*/
void_tree(root);
//printf("%d\n",root);
int sum=max(dp[root][1],dp[root][0]);
printf("%d\n",sum);
return 0;
}