洛谷P1040 加分二叉树(NOIP2003)

树形DP???

题目传送门

简单的树形DP。
f[i][j] 表示从 i j这段节点成为一棵树的最大得分, root[i][j] 表示 i j这段节点取到最大得分时的根节点是谁(中序遍历时要用到)。

因为节点个数足够小,我们可以三维枚举 [l,r] 及根节点。
转移方程:

for (int k=2;k<=n;k++)
    for (int i=1;i<=n-k+1;i++){
        int x=i+k-1;
        for (int j=i;j<=x;j++)
            if (f[i][x]<f[i][j-1]*f[j+1][x]+w[j]){//取最大值
                rt[i][x]=j;
                f[i][x]=f[i][j-1]*f[j+1][x]+w[j];
            }
    }

注意刚开始的值全给1,不然乘的时候就······

代码:

#include<cstdio>
#include<cstring>
#include<algorithm>
#define MAXN 30
using namespace std;
int n;
int f[MAXN+5][MAXN+5],rt[MAXN+5][MAXN+5],w[MAXN+5];
inline char readc(){
    static char buf[100000],*l=buf,*r=buf;
    if (l==r) r=(l=buf)+fread(buf,1,100000,stdin);
    if (l==r) return EOF;
    return *l++;
}
int _read(){//读优(虽然没什么用)
    int num=0; char ch=readc();
    while (ch<'0'||ch>'9') ch=readc();
    while (ch>='0'&&ch<='9') { num=num*10+ch-48; ch=readc(); }
    return num;
}
void dfs(int x,int y){//中序遍历
    if (x<=y){
        printf("%d ",rt[x][y]);
        dfs(x,rt[x][y]-1);
        dfs(rt[x][y]+1,y);
    }
}
int main(){
    n=_read();
    for (int i=0;i<=n;i++)
        for (int j=0;j<=n;j++)
            f[i][j]=1;
    for (int i=1;i<=n;i++){
        w[i]=_read();
        rt[i][i]=i;
        f[i][i]=w[i];
    }
    for (int k=2;k<=n;k++)
        for (int i=1;i<=n-k+1;i++){
            int x=i+k-1;
            for (int j=i;j<=x;j++)
                if (f[i][x]<f[i][j-1]*f[j+1][x]+w[j]){
                    rt[i][x]=j;
                    f[i][x]=f[i][j-1]*f[j+1][x]+w[j];
                }
        }
    printf("%d\n",f[1][n]);
    dfs(1,n);
    return 0;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值