【题目链接】http://www.lydsy.com/JudgeOnline/problem.php?id=1812
【解题思路】转化为二叉树更方便。可以从Bytetown向后dp同时记录每个点在Bytetown方向上的最近锯木厂以满足无后效性。
【呆马】
f[i][j][k]表示第i棵子树,最近锯木厂为j,子树内可再建锯木厂为k
#include<cstdio>
#include<algorithm>
#include<cmath>
#include<cstdlib>
#include<iostream>
#include<cstring>
const int inf=2*(1e9);
using namespace std;
struct st{int to,next,v;} e[101];
int n,cnt,k,x,y,i,d[101],v[101],f[101][101][51],l[101],r[101],fi[101];
void add(int x,int y,int z){e[++cnt].to=y; e[cnt].next=fi[x]; e[cnt].v=z; fi[x]=cnt;}
void dfs(int x)
{
for (int i=fi[x];i;i=e[i].next)
{
d[e[i].to]=d[x]+e[i].v;
dfs(e[i].to);
}
}
int dp(int i,int j,int k)
{
if (f[i][j][k]>=0) return f[i][j][k];
f[i][j][k]=inf;
for (int p=0;p<=k;p++)
{
int t=v[i]*(d[i]-d[j]);
if (l[i]) t+=dp(l[i],j,p);
if (r[i]) t+=dp(r[i],j,k-p);
f[i][j][k]=min(f[i][j][k],t);
if (p<k)
{
t=0;
if (l[i]) t+=dp(l[i],i,p);
if (r[i]) t+=dp(r[i],j,k-1-p);
f[i][j][k]=min(f[i][j][k],t);
}
}
return f[i][j][k];
}
int main()
{
scanf("%d%d\n",&n,&k);
for (i=1;i<=n;i++)
{
scanf("%d%d%d\n",&v[i],&x,&y);
add(x,i,y);
if (!l[x]) l[x]=i;
else r[i]=l[x],l[x]=i;
}
dfs(0);
memset(f,-1,sizeof(f));
printf("%d",dp(0,0,k));
}