树形动态规划 ~~没有上司的晚会

题目链接：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]));
}

路途中。。。。