Ural 1018 Binary Apple Tree(树形dp)

题意:一棵二叉苹果树,每个树枝上都有苹果,求剪枝后剩下Q根树枝的时候的最大苹果数量。(剪枝的时候不能连根拔起)

思路:TreeDP。先由题给出的关系建立一棵二叉树,把边上的苹果移动到子结点,然后对树进行自底而上的DP,状态转移方程式为:f[root][j]=max(f[root.lc][i]+f[root.rc][j-1-i])+val[root],maxx);意思就是说当根root要保留j个点=root的左孩子保留i个点的最大苹果数+root的右孩子保留j-1-i个点的最大苹果数+root点的苹果数。

#include
#include
#include
using namespace std;
const int MAXN=103;
typedef struct
{
    int lc,rc;
    int val;
}Node;
Node tree[MAXN];
int m,n;
int arr[MAXN][MAXN];
int f[MAXN][MAXN];
int vis[MAXN];
void Create(int root)
{
    vis[root]=1;
    for(int i=1;i<=m;i++)
    {
        if(!vis[i]&&arr[root][i])
        {
            if(!tree[root].lc)
            tree[root].lc=i;
            else tree[root].rc=i;
            tree[i].val=arr[root][i];
            Create(i);
        }
    }
}
int TreeDP(int root,int e)
{
    if(!root||!e)return 0;
    if(f[root][e])return f[root][e];
    int maxx=0;
    int l,r;
    for(int i=0;i


你可能感兴趣的:(刷题之旅,dp动态规划,Ural)