二叉苹果树（树形DP）

Posted 2021-01-13 cjoier_gjh的博客

题面

原题

Solution

显然我们考虑这道题目可以很蠢的写一个Dp对吧。
考虑(dp_i,j)表示以i为根，保留j个节点的最大苹果数。
然后就可以树形Dp的写了。

#include<stdio.h>
#include<stdlib.h>
#include<string.h>
#include<math.h>
#include<algorithm>
#include<iostream>
#include<queue>
#define ll long long
#define file(a) freopen(a".in","r",stdin)//;freopen(a".out","w",stdout)
using namespace std;
inline int gi(){
    int sum=0,f=1;char ch=getchar();
    while(ch>'9' || ch<'0'){if(ch=='-')f=-f;ch=getchar();}
    while(ch>='0' && ch<='9'){sum=(sum<<3)+(sum<<1)+ch-'0';ch=getchar();}
    return f*sum;
}
inline ll gl(){
    ll sum=0,f=1;char ch=getchar();
    while(ch>'9' || ch<'0'){if(ch=='-')f=-f;ch=getchar();}
    while(ch>='0' && ch<='9'){sum=(sum<<3)+(sum<<1)+ch-'0';ch=getchar();}
    return f*sum;
}
const int maxn=110,maxq=110;
int dp[maxn][maxq],to[maxn<<1],nxt[maxn<<1],front[maxn],cnt,val[maxn],ans,son[maxn],fa[maxn],n,Q,w[maxn<<1],Son[maxn][3];
void Add(int u,int v,int W){
    to[++cnt]=v;w[cnt]=W;nxt[cnt]=front[u];front[u]=cnt;
}
void dfs(int u,int f){
    fa[u]=f;
    for(int i=front[u];i;i=nxt[i]){
        int v=to[i];
        if(v==f)continue;
        dfs(v,u);val[v]=w[i];
        son[u]++;Son[u][son[u]]=v;
    }
}
int tree_dp(int u,int q){
    if(!q)return dp[u][q]=0;
    if(!son[u])return val[u];
    if(dp[u][q]>-1)return dp[u][q];
    for(int j=0;j<q;j++)
        dp[u][q]=max(dp[u][q],tree_dp(Son[u][1],j)+tree_dp(Son[u][2],q-j-1)+val[u]);
    return dp[u][q];
}
int main(){
#ifndef ONLINE_JUDGE
    file("example");
#endif
    int i,j,m,k;
    memset(dp,-1,sizeof(dp));
    n=gi();Q=gi();Q++;
    for(i=1;i<n;i++){
        int u=gi(),v=gi(),W=gi();
        Add(u,v,W);Add(v,u,W);
    }
    dfs(1,1);
    printf("%d
",tree_dp(1,Q));
    return 0;
}