P2015 二叉苹果树 (树形dp

#include <bits/stdc++.h>
using namespace std;
using VI = vector<int>;
using ll = long long;
using PII = pair<int,int>;
const int mod = 19650827;
int n,q;
int tree[200][2];
int val[200];
int dp[200][200];
bool is_son[200];
int idx = 0;

void dfs(int p){
    dp[p][0] = 0;
    dp[p][1] = val[p];
    if(tree[p][0]) dfs(tree[p][0]),dfs(tree[p][1]);
    else return;
    for(int i=2;i<=n;i++){
        for(int j=0;j<i;j++){
            dp[p][i] = max(dp[p][i],dp[tree[p][1]][j] + dp[tree[p][0]][i-j-1] + val[p]);
        }
    }
}


int main(){
    cin>>n>>q;
    for(int i=1;i<n;i++){
        int a,b,c;
        cin>>a>>b>>c;
        if(is_son[b]){
            swap(a,b);
        }else{
            is_son[b] = true;
        }
        if(!tree[a][0]) tree[a][0] = b;
        else tree[a][1] = b;
        val[b] = c;
    }
    memset(dp,-0x3f,sizeof dp);
    dfs(1);
    cout<<dp[1][q+1];

}

其实像一个树上分组背包问题，只是由于只有俩个种类，也不用跑三重循环了

将数值上的苹果 放到节点上去，然后跑留下几个节点，答案就是dp[1][q+1]