IOI2005 river treedp

一道有趣的题，很容易就看出是treedp，但是在k个伐木场如何分配给多个儿子的问题上纠结不堪，最后发现，如果将k个伐木场看做是空间，每个儿子分配p个伐木场分别得到的最小代价看做权值，这个分配过程其实就相当于一次背包问题，但因为状态设计的不好这样做背包很蛋疼，就照到左儿子右兄弟的思路写了一遍（很早就想写了，今天练练手），其实这个左儿子右兄弟的思路就相当于做了一次背包，基本一样的，可以用一次n^2*k的背包代替。
代码：

#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <iostream>
#include <algorithm>
#define maxn 120000
using namespace std;
int n,k;
int w[maxn],d[maxn];
int chn[maxn];
int lc[maxn],rc[maxn];
int dis[maxn];
inline void dfs(int v){
	int p=lc[v];
	while (p){
		dis[p]=dis[v]+d[p];
		dfs(p);
		p=rc[p];
		}
}
int f[200][200][200];
int get(int i,int j,int rt){
	if (f[i][j][rt]!=f[k+1][k+1][k+1]) return f[i][j][rt];
	f[i][j][rt]=0x7fffffff;
	for (int k=0;k<=j;k++){
		int tmp=w[i]*(dis[i]-dis[rt]);
		if (lc[i]) tmp+=get(lc[i],k,rt);
		if (rc[i]) tmp+=get(rc[i],j-k,rt);
		f[i][j][rt]=min(f[i][j][rt],tmp);
		if (k<j){
			tmp=0;
			if (lc[i]) tmp+=get(lc[i],k,i);
			if (rc[i]) tmp+=get(rc[i],j-k-1,rt);
			f[i][j][rt]=min(tmp,f[i][j][rt]);
			}
		}
	return f[i][j][rt];
}
int main(){
	freopen("river.in","r",stdin);
	freopen("river.out","w",stdout);
	scanf("%d%d",&n,&k);
	for (int i=1;i<=n;i++){
		int p;
		scanf("%d%d%d",&w[i],&p,&d[i]);
		if (!chn[p])
			lc[p]=i;
		else rc[chn[p]]=i;
		chn[p]=i;
		}
	dfs(0);
	memset(f,-1,sizeof(f));
	printf("%d",get(0,k,0));
	return 0;
}