LuoguP2015 二叉苹果树 树形dp

这道题被我秒了是我太强了还是这道题太水了

苹果在树枝上，然后用子树更新节点的f数组即可。

code

#include<iostream>
#include<cstdio>
#include<cstring>
#include<vector>
using namespace std;

const int Maxn = 110;

struct Edge{
    int to,wi,ne;
}edges[Maxn<<1];

int f[Maxn][Maxn];
int size[Maxn],wi[Maxn],first[Maxn];
int q,n,x,y,z,cnte = 1;

void add_edge(int fr,int to,int wi){
    edges[++cnte] = (Edge){to,wi,first[fr]};
    first[fr] = cnte;
}

void dp(int rt,int fa){
    size[rt] = 0;
    for(int i = first[rt];i;i = edges[i].ne){
        int u = edges[i].to;
        if(u != fa){
            dp(u,rt);
            size[rt] += size[u]+1;
            for(int j = q;j > 0;j--)
                for(int k = 0;k <= size[u]&&j-k-1 >= 0;k++)
                    f[rt][j] = 
                        max(f[rt][j],f[u][k]+f[rt][j-k-1]+edges[i].wi);
        }
    }
}

int main(){
    cin >> n >> q;
    for(int i = 1;i < n;i++){
        cin >> x >> y >> z;
        add_edge(x,y,z);
        add_edge(y,x,z);
    }
    dp(1,0);
    cout << f[1][q];
return 0;
}