题目大意
给你一颗二叉树,每一个节点都有一个价值,请你走m条路,使得价值最大。
思路
和二叉苹果树一样。
代码(无注释)
#include<algorithm>
#include<iostream>
#include<cstring>
#include<cstdio>
#include<cmath>
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;
}