rivers ioi2005 树形dp

说句实话，写完这道题，很想吐一口血出来，以示我心情的糟糕；

题目很简单，树形dp，正常做30分钟，硬是做了好几个小时，真是伤心。

题解不写了，只是吐个槽，网上没有用背包写的dp，全是左儿子右兄弟写法，没法对照一下，难受啊。

代码：

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
#define LL long long
int n,K;
const int maxn=210;
const LL inf=3000000000LL;
struct node{
    int y,next,v;
}e[maxn];
int linkk[maxn],len=0,w[maxn],d[maxn][maxn],siz[maxn],q[maxn],tail=0,fa[maxn],ru[maxn],vis[maxn];
int read(){
    int x=0;char ch=getchar();bool flag=0;
    while(ch<'0'||ch>'9'){if(ch=='-')flag=1;ch=getchar();}
    while(ch<='9'&&ch>='0'){x=x*10+ch-'0';ch=getchar();}
    return flag?-x:x;
}
void insert(int x,int y,int v){
    e[++len].y=y;
    e[len].v=v;
    e[len].next=linkk[x];
    linkk[x]=len;
}
void print(int x){printf("%d
",x);}
void print(int x,int y){printf("%d %d
",x,y);}
void init(){
    int y,v;
    n=read(),K=read();
    memset(d,10,sizeof(d));
    for(int i=1;i<=n;i++){
        w[i]=read(),y=read(),v=read();
        insert(i,y,v);insert(y,i,v);
        d[i][y]=v,d[y][i]=v;
    }
    
}
void findd(){
    for(int i=1;i<=n;i++)d[i][i]=0;
    for(int k=0;k<=n;k++)
        for(int i=0;i<=n;i++)
            for(int j=0;j<=n;j++)
                if(d[i][k]+d[k][j]<d[i][j])d[i][j]=d[i][k]+d[k][j];
}
LL f[110][60][110];
void dfs(int x,int f){
    siz[x]=1;
    fa[x]=f;
    for(int i=linkk[x];i;i=e[i].next){
        if(e[i].y==f)continue;
        dfs(e[i].y,x);
        siz[x]+=siz[e[i].y];
        ru[x]++;
    }
    if(siz[x]==1)q[++tail]=x;
}
void work(){
    memset(f,-1,sizeof(f));
    findd();
    dfs(0,0);
    for(int head=1;head<=tail;head++){
        int x=q[head];
        ru[fa[x]]--;
        if(ru[fa[x]]==0)q[++tail]=fa[x];
        LL g[56];
        for(int k=0;k<=siz[x]&&k<=K;k++){
            for(int prev=fa[x];prev!=-1;prev=fa[prev]){
                
                for(int p=0;p<=k;p++)g[p]=0;
                for(int i=linkk[x];i;i=e[i].next){
                    if(e[i].y==fa[x])continue;
                    LL minn=inf;
                    for(int j=k;j>=0;j--){
                        minn=inf;
                        for(int l=0;l<=siz[e[i].y]&&l<=j;l++)
                            minn=min(minn,g[j-l]+f[e[i].y][l][prev]);
                        g[j]=minn;
                    }
                }
                LL ans=g[k]+w[x]*d[prev][x];
                for(int p=0;p<=k;p++)g[p]=0;
                for(int i=linkk[x];i;i=e[i].next){
                    if(e[i].y==fa[x])continue;
                    LL minn=inf;
                    for(int j=k-1;j>=0;j--){
                        minn=inf;
                        for(int l=0;l<=siz[e[i].y]&&l<=j;l++)
                            minn=min(minn,g[j-l]+f[e[i].y][l][x]);
                        g[j]=minn;
                    }
                }
                f[x][k][prev]=min(ans,k-1<0?inf:g[k-1]);
                if(!prev)break;
            }
        }
    }
    cout<<f[0][K][0]<<endl;
}
int main(){
    freopen("1.in","r",stdin);
    freopen("1.out","w",stdout);
    init();
    work();
}