树形DP——洛谷P2015 二叉苹果树

P2015 二叉苹果树

树形DP入门题
f[i][j]代表以i为根的书上保留j个结点的最大权值和
apple[i]表示到根节点i的树枝的苹果数量
核心状态转移方程：
f[i][j]=max(f[i][j],dp(l[i],k)+dp(r[i],j-1-k)+apple[i]);
细节：由于树枝上的苹果数量可能为0，故mp数组初始化为-1

#include<bits/stdc++.h>
using namespace std;
int n,q,mp[105][105],r[105],l[105],apple[105],f[105][105];
void build(int x){
    for(int i=1;i<=n;i++){
        if(mp[x][i]!=-1){
            l[x]=i;
            apple[i]=mp[x][i];
            mp[x][i]=mp[i][x]=-1;
            build(i);
            break;
        }
    }

    for(int i=1;i<=n;i++){
        if(mp[x][i]!=-1){
            r[x]=i;
            apple[i]=mp[x][i];
            mp[x][i]=mp[i][x]=-1;
            build(i);
            break;
        }
    }
}

int dp(int i,int j){
    if(!j)return 0;
    if((!l[i]&&!r[i]))return apple[i];
    if(f[i][j]) return f[i][j];
    for(int k=0;k<=j-1;k++){
        f[i][j]=max(f[i][j],dp(l[i],k)+dp(r[i],j-1-k)+apple[i]);
    }
    return f[i][j];
}

int main(){
    ios::sync_with_stdio(0);
    memset(mp,-1,sizeof(mp));
    cin>>n>>q;q++;
    int u,v,w;
    for(int i=1;i<n;i++){
        cin>>u>>v>>w;
        mp[u][v]=mp[v][u]=w;
    }
    build(1);
    cout<<dp(1,q)<<endl;
    return 0;
}