P1040 [NOIP2003 提高组] 加分二叉树

题目描述
设一个 nn 个节点的二叉树 \text{tree}tree 的中序遍历为(1,2,3,\ldots,n)(1,2,3,…,n)，其中数字 1,2,3,\ldots,n1,2,3,…,n 为节点编号。每个节点都有一个分数（均为正整数），记第 ii 个节点的分数为 d_id
i
​
，\text{tree}tree 及它的每个子树都有一个加分，任一棵子树 \text{subtree}subtree（也包含 \text{tree}tree 本身）的加分计算方法如下：

\text{subtree}subtree 的左子树的加分 \times× \text{subtree}subtree 的右子树的加分 ++ \text{subtree}subtree 的根的分数。

若某个子树为空，规定其加分为 11，叶子的加分就是叶节点本身的分数。不考虑它的空子树。

试求一棵符合中序遍历为 (1,2,3,\ldots,n)(1,2,3,…,n) 且加分最高的二叉树 \text{tree}tree。要求输出

\text{tree}tree 的最高加分。

\text{tree}tree 的前序遍历。

输入格式
第 11 行 11 个整数 nn，为节点个数。

第 22 行 nn 个用空格隔开的整数，为每个节点的分数

输出格式
第 11 行 11 个整数，为最高加分（Ans \le 4,000,000,000Ans≤4,000,000,000）。

第 22 行 nn 个用空格隔开的整数，为该树的前序遍历。

区间DP，值得记录

#include<bits/stdc++.h>
using namespace std;
#define ll long long
ll n,a[50],dp[50][50],root[50][50];
void print(int l,int r){
    if(l>r)return;
    int rt=root[l][r];
    cout<<rt<<' ';
    print(l,rt-1);
    print(rt+1,r);
}

int main(){
    ios::sync_with_stdio(0);
    cin>>n;
    for(int i=1;i<=n;i++){
        cin>>a[i];
        dp[i][i]=a[i];
        root[i][i]=i;

    }

    for(int l=1;l<n;l++){
        for(int i=1;i+l<=n;i++){
            int j=i+l;

            if(dp[i+1][j]+a[i]>=dp[i][j-1]+a[j]){
                dp[i][j]=dp[i+1][j]+a[i];
                root[i][j]=i;
            }
            else{
                dp[i][j]=dp[i][j-1]+a[j];
                root[i][j]=j;
            }
            for(int k=i+1;k<j;k++){
                if(dp[i][j]<dp[i][k-1]*dp[k+1][j]+a[k]){
                    dp[i][j]=dp[i][k-1]*dp[k+1][j]+a[k];
                    root[i][j]=k;
                }
            }
        }
    }



    /*for(int i=1;i<=n;i++){
        for(int j=1;j<=n;j++){
            cout<<dp[i][j]<<' ';
        }
        cout<<endl;
    }*/


    cout<<dp[1][n]<<endl;
    print(1,n);
    return 0;
}