BZOJ 1812: [Ioi2005]riv

树形动规

直接抄了论文里的写法，好慢的说。。。。。。。

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
struct Edge{int to,next;}e[205];
int head[105],cnt;
void ins(int u,int v){
	e[++cnt]=(Edge){v,head[u]};head[u]=cnt;
}
int w[105],pa[105],p[105],move[105][105]; 
int f[105][105][55];
bool vis[105][105][55];
void pre(int u){
	int t=u;
	while(pa[t]>=0){
		move[u][pa[t]]=move[u][t]+w[u]*p[t];
		t=pa[t];
	}
	t=u;
	for(int i=head[u];i;i=e[i].next){
		pre(e[i].to);
		w[u]+=w[e[i].to];
		f[u][u][0]+=f[e[i].to][u][0];
	}
	vis[u][t][0]=1;
	while(pa[t]>=0){
		vis[u][pa[t]][0]=1;
		f[u][pa[t]][0]=f[u][t][0]+w[u]*p[t];
		t=pa[t];
	}
}
int dp(int u,int to,int k){
	if(vis[u][to][k])return f[u][to][k];
	vis[u][to][k]=1;
	int h1[55],h2[55];
	memset(h1,0,sizeof(h1));
	for(int i=0;i<=k;i++)h2[i]=move[u][to];
	for(int i=head[u];i;i=e[i].next){
		int v=e[i].to;
		for(int j=k;j>=0;j--){
			h1[j]+=dp(v,u,0);
			h2[j]+=dp(v,to,0);
			for(int t=1;t<=j;t++){
				h1[j]=min(h1[j],h1[j-t]+dp(v,u,t));
				h2[j]=min(h2[j],h2[j-t]+dp(v,to,t));
			}
		}
	}
	f[u][to][k]=min(h1[k-1],h2[k]);
	return f[u][to][k];
}
int main(){
	//freopen("a.in","r",stdin);
	//freopen("a.out","w",stdout); 
	int n,k;scanf("%d%d",&n,&k);
	for(int i=1;i<=n;i++){
		scanf("%d%d%d",&w[i],&pa[i],&p[i]);
		ins(pa[i],i);
	}
	pa[0]=-1;
	pre(0);
	printf("%d\n",dp(0,0,k));
	return 0;
}