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<<];
     int lc[maxn],rc[maxn],v[maxn];
     int first[maxn],le;
     Tree() {memset(first,,sizeof(first));le=;}
     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]=;
         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][]=;
         for (int i=;i<=Q;i++)
         {
             f[x][i]=max(f[lc[x]][]+f[rc[x]][i-]+v[rc[x]],f[lc[x]][i-]+f[rc[x]][]+v[lc[x]]);
             for (int j=;j<=i-;j++)
                 f[x][i]=max(f[x][i],f[lc[x]][j-]+f[rc[x]][i-j-]+v[lc[x]]+v[rc[x]]);
         }
     }
 }t;
 int x,y,v;
 int main()
 {
     scanf("%d%d",&n,&Q);
     for (int i=;i<n;i++)
     {
         scanf("%d%d%d",&x,&y,&v);
         t.insert(x,y,v);
     }
     t.dfs(,);
     t.dp();
     printf("%d\n",f[][Q]);
     return ;
 }