树结构和DP的结合
一般分为以下几个步骤
1.建立树(通常为有根树)
2.遍历树(可以结合DFS也可以用记忆化搜索)
3.DP
通常使用的DP状态dp[i][0],dp[i][1]代表了第i个节点选用和不选用的状态。
题目地址
以HDU1520为例,说明DP状态
设dp[i][j]表示这个节点选用或不选用的最优解
我们发现每个节点都有选和不选的状态,如果第i个节点选了那么就有
dp[i][1]+=dp[i+1][0];
否则
dp[i][0]+=max(dp[i+1][1],dp[i+1][0]);
其中我们视i+1为i的son节点
#include <bits/stdc++.h>
using namespace std;
int value[6010],dp[6010][2],father[6010];
vector <int> son[6010];
void DP(int t)
{
dp[t][0]=0;
dp[t][1]=value[t];
for(int i=0;i<son[t].size();i++)
{
int son1=son[t][i];
DP(son1);//此处注意,顺序不能在下面两个dp前面,毕竟你只知道 dp[t][0]=0; dp[t][1]=value[t];
//你需要知道他的子节点的值,实际上会一直dfs到这条树的最低端。
dp[t][0]+=max(dp[son1][1],dp[son1][0]);
dp[t][1]+=dp[son1][0];
}
}
int main()
{
freopen("in.txt","r",stdin);
int n;
cin>>n;
for(int i=1;i<=n;i++)
{
cin>>value[i];
father[i]=-1;//记住初始化节点关系
son[i].clear()//这句其实没有也不影响,但是思路上来讲是需要的。
}
int a,b;
while(cin>>a>>b&&a!=0&&b!=0)
{
son[b].push_back(a);//因为一个父节点有多个子节点,一个子节点有1个父节点,所以用这种方式
father[a]=b;
}
int t=1;
while(father[t]!=-1)
t=father[t];
DP(t);
cout<<max(dp[t][0],dp[t][1])<<endl;
return 0;
}