二叉苹果树

题目描述

有一棵苹果树，如果树枝有分叉，一定是分2叉（就是说没有只有1个儿子的结点）
这棵树共有N个结点（叶子点或者树枝分叉点），编号为1-N,树根编号一定是1。
我们用一根树枝两端连接的结点的编号来描述一根树枝的位置。现在这颗树枝条太多了，需要剪枝。但是一些树枝上长有苹果。
给定需要保留的树枝数量，求出最多能留住多少苹果。
输入格式：
第1行2个数，N和Q(1<=Q＜＝ N,1＜N＜＝100)。
N表示树的结点数，Q表示要保留的树枝数量。接下来N-1行描述树枝的信息。
每行3个整数，前两个是它连接的结点的编号。第3个数是这根树枝上苹果的数量。
每根树枝上的苹果不超过30000个。
输出格式：
一个数，最多能留住的苹果的数量。

直接复制一下我在洛谷上发的题解好了0.0

这道题可以把边权转化为点权来做，然后dp找出深度。状态转移方程就是枚举一个j是以u为根的子树中选了几个点。以u为根，u是一定要选的，所以我们要从他的子树中选出j-1个点，枚举从左子树中选了k个点来更新状态。具体见代码。

#include<bits/stdc++.h>
using namespace std;
const int MAXN=205;
int head[MAXN],cnt=0;
struct edge{
    int to,next,w;
}e[MAXN<<1];
inline void add(int u,int v,int w){e[++cnt]=(edge){v,head[u],w},head[u]=cnt;}
int n,m,tem1,tem2,tem3,dw[MAXN],size[MAXN],dp[MAXN][MAXN],lson[MAXN<<5],rson[MAXN<<5]; //开够大小 
void dfs1(int u,int fa){
    size[u]=1;
    for(int i=head[u];i;i=e[i].next){
        int v=e[i].to;
        if(v==fa)continue;
        dw[v]=e[i].w;
        if(!lson[u])lson[u]=v;
        else rson[u]=v;
        dfs1(v,u);
        size[u]+=size[v];
    }
}
//dp[i][j]=max(dp[i][j],dp[lson][j-1-k]+dp[rson][k]+w[i])   0<=k<=j-1   状态转移 
void dfs2(int u,int fa){
    for(int i=head[u];i;i=e[i].next){
        int v=e[i].to;
        if(v==fa)continue;
        dfs2(v,u);
        for(int j=2;j<=size[u];j++){
            for(int k=0;k<=j-1;k++){
                dp[u][j]=max(dp[u][j],dp[lson[u]][j-1-k]+dp[rson[u]][k]+dw[u]);//转移 
            }
        }
    }
}
int main(){
    memset(dw,0,sizeof(dw));
    scanf("%d%d",&n,&m);
    for(int i=1;i<n;i++){
        scanf("%d%d%d",&tem1,&tem2,&tem3);
        add(tem1,tem2,tem3);
        add(tem2,tem1,tem3);
    }
    dfs1(1,1);//预处理 
    for(int i=1;i<=n;i++){
        dp[i][1]=dw[i];
    }
    dfs2(1,1);
    printf("%d",dp[1][m+1]);
    return 0;
}