树形DP入门

树形DP都有啥？（谁说对了就教他）

ANSWER：在树上跑的DP……（逃）。

（别打我）事实上，树形DP本质上就是一般的DP，不同的，他把链上动态规划状态拓展成平面上的树；

　　因此，他的策略应该跟一般的DP一样的说。

　　　　当然，他的方向毕竟要跟链表不同，一般的树上DP多是从叶子或给定的根开始你的状态。

直接上例题！

P2015 二叉苹果树（板子） 题目链接：https://www.luogu.org/problem/P2015

 

一句话题意：给定一棵树，求出上面权值最大的树的权值。

了解到如果把这个边保留下来，那么他的父点以上向根连的边必然也选。

　　因此我们可以从叶子向上拓展子树进行动态规划。

　　　　设dp[u][j]表示以u为根，选了j个边的最大权值；

　　设计出状态：dp[u][j]=max(dp[u][j],dp[u][j-k-1]+dp[v][k]+ed[i].val);

　　　　　　其中后半部分表示砍去。

剩下的就是建树与建边得斯！(￣▽￣)／

代码如下：

#include<bits/stdc++.h>//动规 
using namespace std;
#define maxn 1005
struct edge{
    int nxt,to,val;
}ea[maxn];
int n,cnt,sum,head[maxn],dp[maxn][maxn],en[maxn],q,k;
bool vis1[maxn],vis2[maxn];
inline int read(){
    int a=1,b=0;char t;
    while(t<'0'||t>'9'){
        if(t=='-') a=-1;
        t=getchar();
    }
    while(t>='0'&&t<='9') b=b*10+t-'0',t=getchar();
    return a*b;
}
inline void write(int x){
    if(!x) putchar('0');
    else{
        int dig[35],top=0;
        while(x) dig[++top]=x%10,x/=10;
        while(top) putchar(dig[top--]+'0');
    }
}
void add_edge(int u,int v,int z){
    ea[++cnt].to=v,ea[cnt].nxt=head[u],head[u]=cnt,ea[cnt].val=z;
    ea[++cnt].to=u,ea[cnt].nxt=head[v],head[v]=cnt,ea[cnt].val=z;
}
void dfs(int u,int fa){
    for(int i=head[u];i;i=ea[i].nxt){
        int v=ea[i].to;
        if(v!=fa){
            dfs(v,u);
            en[u]+=en[v]+1;
            for(int j=min(en[u],q);j;j--)
                for(int k=min(j-1,en[v]);k>=0;k--)//手残顺手打了一个k，应该是k>=0（k==0刚好到） 
                    dp[u][j]=max(dp[u][j],dp[u][j-k-1]+dp[v][k]+ea[i].val);
        }
    }
}
int main(){
    n=read(),q=read();
    for(int i=1,u,v,z;i<n;i++){
        u=read(),v=read(),z=read();
        add_edge(u,v,z);
    }
    dfs(1,-1);
    write(dp[1][q]);
}