[CODEVS1090][NOIP2013]加分二叉树（树形dp）

来源：互联网发布：linux是什么系统编辑：程序博客网时间：2024/05/29 13:21

题目描述

传送门

题解

f[i][j][k]表示节点j～k以i为根的最大加分。
用记忆化搜索来搞。
要记录一下每一个区间的根，然后递归输出。

代码

#include<iostream>#include<cstring>#include<cstdio>using namespace std;#define LL long longconst int N=35;int n; LL ans;int root[N][N],ansroot[N][N];LL score[N],f[N][N][N];inline LL dfs(int now,int l,int r){    if (f[now][l][r]!=-1) return f[now][l][r];    if (l==r){        f[now][l][r]=score[l];        root[l][r]=l;        return f[now][l][r];    }    LL lch=0,rch=0,t;    for (int i=l;i<now;++i)      if (lch<(t=dfs(i,l,now-1))){        lch=t;        root[l][now-1]=i;      }    for (int i=now+1;i<=r;++i)      if (rch<(t=dfs(i,now+1,r))){        rch=t;        root[now+1][r]=i;      }    if (!(l<=now-1)) lch=1; if (!(now+1<=r)) rch=1;    f[now][l][r]=lch*rch+score[now];    return f[now][l][r];}inline void print(int l,int r){    int now=ansroot[l][r];    printf("%d ",now);    if (l<=now-1) print(l,now-1);    if (now+1<=r) print(now+1,r);}int main(){    scanf("%d",&n);    for (int i=1;i<=n;++i) scanf("%I64d",&score[i]);    memset(f,-1,sizeof(f));    for (int i=1;i<=n;++i){        LL t=dfs(i,1,n);        if (t>ans){            ans=t; root[1][n]=i;            for (int j=1;j<=n;++j)              for (int k=j;k<=n;++k)                ansroot[j][k]=root[j][k];        }    }    printf("%I64d\n",ans);    print(1,n);}

0 0