SSL1565 将功补过【树形DP】

题目大意

给你一颗二叉树，每一个节点都有一个价值，请你走m条路，使得价值最大。

思路

和二叉苹果树一样。

代码（无注释）

#include
#include
#include
#include
#include
using namespace std;
int jz[5010][5010],tree[5010][3],f[5010][5010];
int num[501000];
int n,q,x,y,w,m;
void mt(int x);

void build(int x,int y,int js)
{
	num[y]=jz[x][y];
	tree[x][js]=y;
	jz[x][y]=-1,jz[y][x]=-1;
	mt(y);
}
void mt(int x)
{
	int js=0;
	for(int i=1; i<=n; i++)
	 {
	 	if(jz[x][i]>=0)
	 	 {
	 	 	js++;
	 	 	build(x,i,js);
	 	 	if(js==2)
			  return;
	 	 }
	 }
}
void dfs(int x,int m)
{
	if(m==0)
	  f[x][m]=0;
	else if(tree[x][1]==0&&tree[x][2]==0)
	  f[x][m]=num[x];
	else
	 {
	 	for(int i=0; i<m; i++)
	 	 {
	 	 	if(f[tree[x][1]][i]==0)
	 	 	  dfs(tree[x][1],i);
	 	 	if(f[tree[x][2]][m-i-1]==0)
	 	 	  dfs(tree[x][2],m-i-1);
	 	 	f[x][m]=max(f[x][m],f[tree[x][1]][i]+f[tree[x][2]][m-i-1]+num[x]);
	 	 }
	 }
}
int main()
{
    cin>>n>>q>>m;
    for(int i=0; i<=n; i++)
     for(int j=0; j<=n; j++)
        jz[i][j]=-1;
    for(int i=1; i<n; i++)
     {
     	scanf("%d%d%d",&x,&y,&w);
     	jz[x][y]=w;
     	jz[y][x]=w;
     }
    mt(1);
    dfs(1,q+1);
    if(f[1][q+1]>=m)
      cout<<"TRUE"<<endl<<f[1][q+1]-m;
    else
      cout<<"FALSE"<<endl<<f[1][q+1]-m;
    return 0;
}