P1273 有线电视网 ( 树上背包

#include<bits/stdc++.h>
using namespace std;
using VI = vector<int>;
int n,m;
VI g[3010];
VI gv[3010];
int v[3010];
int dp[3010][3010];
//第i个节点 
int pd[3010];
int siz[3010];

int dfs(int x){
    dp[x][0] = 0;
    if(pd[x]){
        dp[x][1] = v[x];
        siz[x] = 1;
        return 1;
    }

    for(int i = 0 ; i < g[x].size() ; i++){
        int to = g[x][i];
        int cost = gv[x][i];
        siz[x] += dfs(to);
        for(int j = siz[x] ; j >= 0; j--){
            for(int k = 0 ; k <= siz[to] ; k++){
                if(j - k >= 0) dp[x][j] = max(dp[x][j] , 
                dp[x][j-k] + dp[to][k] - cost);
            }
        }
    }
    

    return siz[x];
    
}




int main(){
    cin>>n>>m;
    for(int i = 1 ; i <= n - m ; i++){
        int k;
        cin>>k;
        for(int j = 1 ; j <= k ; j++){
            int a,c;
            cin>>a>>c;
            g[i].push_back(a);
            gv[i].push_back(c);
        }   
    }

    for(int i = n - m + 1 ; i <= n ; i++) {
        pd[i] = 1;
        cin>>v[i];
    }
    memset(dp, -0x3f , sizeof dp);
    dfs(1);
    for(int i = n ; i; i--){

        if(dp[1][i] >= 0){
            cout<<i;
            break;
        }
    }
    
}