题目链接:https://www.contesthunter.org/Contest/%E7%9C%9F%C2%B7%E6%B0%B4%E9%A2%98%E8%B5%9B%20%EF%BC%832%28Codeforces%29/Problem/Show/%E6%B2%A1%E6%9C%89%E4%B8%8A%E5%8F%B8%E7%9A%84%E6%99%9A%E4%BC%9A
树形动态规划:是指当动态规划的各阶段形成一棵树,利用各阶段之间的关系(动态转移方程),从叶节点(边界)开始逐步向上一层的节点(即父节点)进行动态规划,直到动规到根节点,(即原问题),求的问题的最优解。
按照定义的代码:
#include<iostream>
#include<algorithm>
using namespace std;
int main(){
int qf[201],f[201][2],shs[201],xb[201][201],shu[201][201];
int n,x,s,maxc,j,k,a,b,l,i;
cin>>n;
for(i=0;i<=n;i++){
xb[i][0]=0;
shs[i]=0;
}
for(i=1;i<=n;i++) cin>>qf[i];
l=1;
k=1;
while(l!=0 || k!=0){
cin>>l>>k;
shs[l]=k;
xb[k][0]++;
xb[k][xb[k][0]]=l;
}
maxc=0;
for(i=1;i<=n;i++){
x=i;
s=1;
while(shs[x]!=0){
x=shs[x];
s=s+1;
}
shu[s][0]++;
shu[s][shu[s][0]]=i;
if (s>maxc) maxc=s;
}//建树,maxc存储最大层数
for(i=maxc;i>=1;i--)
for(j=1;j<=shu[i][0];j++){
if(xb[shu[i][j]][0]==0){
f[shu[i][j]][0]=0;
f[shu[i][j]][1]=qf[shu[i][j]];
}
else{
f[shu[i][j]][0]=0;
f[shu[i][j]][1]=qf[shu[i][j]];
for(k=1;k<=xb[shu[i][j]][0];k++){
a=f[xb[shu[i][j]][k]][0];
b=f[xb[shu[i][j]][k]][1];
f[shu[i][j]][1]+=a;
if(b>a)a=b;
f[shu[i][j]][0]+=a;
}//动态转移方程
}
}
s=0;
for(i=1;i<=shu[1][0];i++){
a=f[shu[1][i]][0];b=f[shu[1][i]][1];
if(a<b) a=b;
s=s+a;
}//从顶头上司那里择优选择
cout<<s<<endl;
//system("pause");
return 0;
}
在做题网站上看见一大牛的代码:
不过这个代码适用的范围比较小。
#include<cstdio>
#include<algorithm>
using namespace std;
int n,a[6100],f[6100][2],s; //f[][0]表示不含该点得最优值,f[][1]表示把改点包含进去的最优值。
int p[6100],q[6100];
void work(int x){
for(int i=1;i<n;i++)
if (q[i]==x){
int y=p[i];
work(y);
f[x][1]+=f[y][0];
f[x][0]+=max(f[y][1],f[y][0]); //用回朔求各状态的最优值。
}
}
int main(){
scanf("%d",&n);
s=n*(n+1)/2;
for (int i=1;i<=n;i++){
scanf("%d",&f[i][1]);
f[i][0]=0;
}
for (int i=1;i<n;i++){
scanf("%d%d",&p[i],&q[i]);
s-=p[i]; //求初始点
}
work(s);
printf("%d\n",max(f[s][0],f[s][1]));
}
路途中。。。。