bzoj1812 [Ioi2005]riv（树形dp+背包转移）

类似之前题的套路，还是当前决策影响未来行动的花费。不过当前决策对未来行动的的花费的影响又与未来的状态有关，这时我们假设未来的状态，将不同的影响沿着不同的状态传递到未来。

对于本题而言，子树内一个点是否建了对我根节点的决策有影响，因此我们在决策子节点的时候增设一维状态表示未来，即最近的建了的祖先为j，直接把贡献提前计算了。

背包转移即可。
注意是一定要分配正好j个，所有儿子都要算贡献，因此要不断给tmp清最大值。
复杂度$O(n^2k^2)$

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <vector>
using namespace std;
#define ll long long
#define inf 0x3f3f3f3f
#define N 110
inline char gc(){
    static char buf[1<<16],*S,*T;
    if(S==T){T=(S=buf)+fread(buf,1,1<<16,stdin);if(T==S) return EOF;}
    return *S++;
}
inline int read(){
    int x=0,f=1;char ch=gc();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=gc();}
    while(ch>='0'&&ch<='9') x=x*10+ch-'0',ch=gc();
    return x*f;
}
int n,K,w[N],dis[N],a[N],tot=0,tmp[N],dp[N][51][N],ans=inf;
//dp[x][i][j],x的子树内,建i个,最近的建了的祖先为j的最小花费
vector<int>son[N];
void dfs(int x){
    if(!son[x].size()){
        for(int i=1;i<=tot;++i) dp[x][0][a[i]]=w[x]*(dis[x]-dis[a[i]]);
        dp[x][0][x]=inf;return;
    }a[++tot]=x;
    for(int i=0;i<son[x].size();++i){
        int y=son[x][i];dis[y]+=dis[x];dfs(y);
    }for(int jj=1;jj<tot;++jj){
        for(int ii=0;ii<son[x].size();++ii){
            memset(tmp,inf,sizeof(tmp));int y=son[x][ii];
            for(int i=K;i>=0;--i)
                for(int j=i;j>=0;--j)
                    tmp[i]=min(tmp[i],dp[x][j][a[jj]]+min(dp[y][i-j][a[jj]],dp[y][i-j][y]));
            for(int i=K;i>=0;--i) dp[x][i][a[jj]]=tmp[i];
        }for(int i=K;i>=0;--i) dp[x][i][a[jj]]+=w[x]*(dis[x]-dis[a[jj]]);
    }dp[x][0][x]=inf;
    for(int ii=0;ii<son[x].size();++ii){
        int y=son[x][ii];memset(tmp,inf,sizeof(tmp));
        for(int i=K;i>=0;--i)
            for(int j=i;j>=0;--j)
                tmp[i]=min(tmp[i],dp[x][j][x]+min(dp[y][i-j][x],dp[y][i-j][y]));
        for(int i=K;i>=0;--i) dp[x][i][x]=tmp[i];
    }--tot;
}
int main(){
//  freopen("a.in","r",stdin);
    n=read();K=read();
    for(int i=1;i<=n;++i){
        w[i]=read();int x=read();son[x].push_back(i);dis[i]=read();
    }a[++tot]=0;
    for(int ii=0;ii<son[0].size();++ii){
        int y=son[0][ii];dfs(y);memset(tmp,inf,sizeof(tmp));
        for(int i=K;i>=0;--i)
            for(int j=i;j>=0;--j)
                tmp[i]=min(tmp[i],dp[0][j][0]+min(dp[y][i-j][0],dp[y][i-j][y]));
        for(int i=K;i>=0;--i) dp[0][i][0]=tmp[i];
    }for(int i=K;i>=0;--i) ans=min(ans,dp[0][i][0]);
    printf("%d\n",ans);
    return 0;
}