P2015 二叉苹果树

辣鸡题面

这题目确实是比较经典的树形dp，里面确实还是有很多需要理解的地方

#include<bits/stdc++.h>
using namespace std;
const int N=1005;

int n,q;
int dis[N][N],l[N],r[N],a[N],f[N][N];

int get(){
    char zy=getchar();
    int z=1,y=0;
    while(zy>'9'||zy<'0'){
        if(zy=='-')z=-1;
        zy=getchar();
    }
    while(zy>='0'&&zy<='9'){
        y=y*10+zy-'0';
        zy=getchar();
    }
    return z*y;
}

void build(int u){
    for(int i=1;i<=n;i++){
        if(dis[u][i]>=0){
            if(!l[u]){
                a[i]=dis[u][i];
                dis[u][i]=dis[i][u]=-1;
                l[u]=i;
                build(i);
                continue;
            }
            a[i]=dis[u][i];
            dis[u][i]=dis[i][u]=-1;
            r[u]=i;
            build(i);
            break;
        }
    }
}

int Max(int a,int b){return a>b?a:b;}

int tdp(int u,int t){
    if(t==0) return 0;
    if(l[u]==0&&r[u]==0)return a[u];
    if(f[u][t])return f[u][t];
    for(int i=0;i<t;i++){
        f[u][t]=Max(f[u][t],tdp(l[u],i)+tdp(r[u],t-i-1)+a[u]);
    }
    return f[u][t];
}

int main(){
    n=get();q=get()+1;
    for(int i=1;i<=n;i++){
        for(int j=1;j<=n;j++){
            dis[i][j]=-1;
        }
    }
    for(int i=1;i<n;i++){
        int u=get(),v=get();
        dis[u][v]=dis[v][u]=get();
    }
    build(1);
    printf("%d
",tdp(1,q));
    return 0;
}