bzoj1812[Ioi2005] riv

题目链接：bzoj1812
题目大意：
给你一颗有n个节点的树根为0的树。树根一定是个标记点。让你再在树上标记k个点。使得 每个点的点权乘以它到往上走遇到的第一个标记点的距离 的和最小。

题解：
树形dp+森林转二叉
诶，一直想着平常的树形dp，不知道怎么很好的转移。
膜了发题解。。。转成二叉树，好像方程确实好写了><。
设$f[i][j][k]$表示在$i$这棵子树(新的树)中建$j$个伐木场(就是标那么多个点)，且这棵子树上面最近的标记点(不含$i$)是$k$。
于是方程：
①在$i$建个场：$f[i][j][k]=\min(f[i][j][k],f[lc[i]][p][i]+f[rc[i]][j-p-1][k])$
②不在$i$建场：
$f[i][j][k]=\min(f[i][j][k],f[rc[i]][j-p][k]+f[lc[i]][p][k]+c[i]*dis(i,k)$

#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
#define N 110

const int inf=0x7fffffff;
int f[N][N][N],dis[N],K;
int lc[N],rc[N],c[N],d[N];
int mymin(int x,int y){return (x<y)?x:y;}
void dfs(int x)
{
    for (int y=lc[x];y!=-1;y=rc[y])
    {
        dis[y]=dis[x]+d[y];
        dfs(y);
    }
}
int dp(int x,int j,int k)
{
    if (f[x][j][k]!=-1) return f[x][j][k];
    f[x][j][k]=inf;
    for (int p=0;p<=j;p++)
    {
        if (p!=j)
        {
            int ln=0;
            if (lc[x]!=-1) ln+=dp(lc[x],p,x);
            if (rc[x]!=-1) ln+=dp(rc[x],j-p-1,k);
            f[x][j][k]=mymin(f[x][j][k],ln);
        }
        int ln=c[x]*(dis[x]-dis[k]);
        if (lc[x]!=-1) ln+=dp(lc[x],p,k);
        if (rc[x]!=-1) ln+=dp(rc[x],j-p,k);
        f[x][j][k]=mymin(f[x][j][k],ln);
    }
    return f[x][j][k];
}
int main()
{
    //freopen("a.in","r",stdin);
    //freopen("a.out","w",stdout);
    int n,i,w,x,s;
    scanf("%d%d",&n,&K);n++;
    memset(f,-1,sizeof(f));
    for (i=1;i<=n;i++) lc[i]=rc[i]=-1;
    for (i=2;i<=n;i++)
    {
        scanf("%d%d%d",&w,&x,&s);
        x++;d[i]=s;c[i]=w;
        if (lc[x]==-1) lc[x]=i;
        else
        {
            int p=lc[x];
            while (rc[p]!=-1) p=rc[p];
            rc[p]=i;
        }
    }
    dfs(1);int ans=dp(1,K,1);
    printf("%d\n",ans);
    return 0;
}