P2015 二叉树苹果树

  这是一道树形dp,很简单,代码也很好懂。

  只要明确几个变量的定义:

  f[i][j]:以i为根的树上保留j个结点的最大权值和。

  l[i],r[i]:i结点的左右儿子。

  剩下的没什么了啊……很好懂啊。

  代码提示:

  

#include<cstdio>
#include<iostream>
#include<cstring>
using namespace std;
#define maxn 200
int n,q,map[maxn][maxn],a[maxn],f[maxn][maxn],l[maxn],r[maxn];
void MakeTree(int v)
{
    for(int i=1;i<=n;i++)
    {
        if(map[v][i]>=0)
        {
            a[i]=map[v][i];
            map[v][i]=map[i][v]=-1;
            l[v]=i;
            MakeTree(i);
            break;
        }
    }
    for(int i=1;i<=n;i++)
    {
        if(map[v][i]>=0)
        {
            a[i]=map[v][i];
            map[v][i]=map[i][v]=-1;
            r[v]=i;
            MakeTree(i);
            break;
        }
    }
}
int DP(int i,int j)
{
    if(j==0) return 0;
    if(l[i]==0&&r[i]==0) return a[i];
    if(f[i][j]>0) return f[i][j];
    for(int k=0;k<=j-1;k++)
    f[i][j]=max(f[i][j],DP(l[i],k)+DP(r[i],j-1-k)+a[i]);
    return f[i][j];
}
int main()
{
    scanf("%d%d",&n,&q);
    q++;
    for(int i=1;i<=n;i++)
    for(int j=1;j<=n;j++)
    map[i][j]=-1;
    for(int i=1;i<=n-1;i++)
    {
        int x,y,z;
        scanf("%d%d%d",&x,&y,&z);
        map[x][y]=z;
        map[y][x]=z;
    }
    MakeTree(1);
    printf("%d",DP(1,q));
    return 0;
}

 

posted @ 2018-12-07 22:24  paopo  阅读(222)  评论(0编辑  收藏  举报