CODEVS5565 二叉苹果树

n<=100个点的根为1的二叉树，树边上有苹果，求保留Q<=n条边的最多苹果数。

树形DP，f[i][j]--节点i为根的子树保留j条边最优方案，f[i][0]=0,f[i][j]=max(f[lc[i]][k-1]+f[rc[i]][j-k-1]+v[lc[i]]+v[rc[i]])，这是左右都选的情况，再加只选左只选右方案即可。

#include<stdio.h>
#include<stdlib.h>
#include<string.h>
#include<algorithm>
//#include<iostream>
using namespace std;

int n,Q;
#define maxn 111
int f[maxn][maxn];
struct Edge{int to,next,v;};
struct Tree
{
    Edge edge[maxn<<1];
    int lc[maxn],rc[maxn],v[maxn];
    int first[maxn],le;
    Tree() {memset(first,0,sizeof(first));le=2;}
    void in(int x,int y,int v)
    {
        edge[le].to=y;
        edge[le].v=v;
        edge[le].next=first[x];
        first[x]=le++;
    }
    void insert(int x,int y,int v)
    {
        in(x,y,v);
        in(y,x,v);
    }
    void dfs(int x,int fa)
    {
        lc[x]=rc[x]=0;
        for (int i=first[x];i;i=edge[i].next)
            if (edge[i].to!=fa)
            {
                if (!lc[x]) lc[x]=edge[i].to;
                else rc[x]=edge[i].to;
                v[edge[i].to]=edge[i].v;
                dfs(edge[i].to,x);
            }
    }
    void dp(int x)
    {
        if (lc[x]) dp(lc[x]);
        if (rc[x]) dp(rc[x]);
        f[x][0]=0;
        for (int i=1;i<=Q;i++)
        {
            f[x][i]=max(f[lc[x]][0]+f[rc[x]][i-1]+v[rc[x]],f[lc[x]][i-1]+f[rc[x]][0]+v[lc[x]]);
            for (int j=1;j<=i-1;j++)
                f[x][i]=max(f[x][i],f[lc[x]][j-1]+f[rc[x]][i-j-1]+v[lc[x]]+v[rc[x]]);
        }
    }
}t;
int x,y,v;
int main()
{
    scanf("%d%d",&n,&Q);
    for (int i=1;i<n;i++)
    {
        scanf("%d%d%d",&x,&y,&v);
        t.insert(x,y,v);
    }
    t.dfs(1,0);
    t.dp(1);
    printf("%d
",f[1][Q]);
    return 0;
}