P1040 加分二叉树

P1040 加分二叉树

洛谷题 https://www.luogu.com.cn/problem/P1040
题目描述

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

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

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

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

tree 的前序遍历。

输入格式

第 1行 1个整数 n，为节点个数。

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

输出格式

第 1 行 1个整数，为最高加分（ Ans \le 4,000,000,000Ans≤4,000,000,000）。

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

输入输出样例

输入 #1 复制
5
5 7 1 2 10
输出 #1 复制
145
3 1 2 4 5

相信大家都知道了题目意思，但我还是凑字数吧！题目要求我们用以1–n当做节点(题目给出权值)编号。让我求出其得分最高的二叉树(左节点乘右节点加根，节点没有视为1)。
我们的具体思路由题目给出的该二叉树的中序遍历为1–n(前中后都是以根为准的，比如前序遍历:根左右)。又考虑到乘比加的贡献度高。我们只需要一一枚举其根节点就行。所以我们的具体做法是f[i][j]=max(f[i][j],f[i][k-1]*f[k+1][j]+d[k])
f[i][j]表示在i到j的范围内选一个k作为根，求他们最大的得分就行，学过dp的应该都能理解吧！
从小到大。至于前序遍历用一个数组存最优节点在递归输出应该不难。下面是代码。欢迎指正。有问题可以提。

#include <iostream>
#define ll long long
using namespace std;
ll f[50][50];
int ro[50][50];
void qxsc(int l,int r)//求前序遍历
{
    if(l==r) {cout<<ro[l][r]<<" ";return;}
    if(l>r) return ;
    cout<<ro[l][r]<<" ";
    qxsc(l,ro[l][r]-1);
    qxsc(ro[l][r]+1,r);
}
int main()
{
    int n;
    cin>>n;
    for(int i=1;i<=n;++i)
        cin>>f[i][i],ro[i][i]=i;//初始化默认节点
    for(int len=1;len<n;++len)
        for(int i=1;i+len<=n;++i)
    {
        int j=i+len;
       f[i][j]=f[i+1][j]+f[i][i];//默认左节点为空。
       ro[i][j]=i;//最开始默认最优。
        for(int k=i+1;k<j;++k)
           if(f[i][j]<f[i][k-1]*f[k+1][j]+f[k][k])
        {
            f[i][j]=f[i][k-1]*f[k+1][j]+f[k][k];
            ro[i][j]=k;//存最优根或节点的值。
        }
    }
    cout << f[1][n] << endl;
    qxsc(1,n);
    return 0;
}

额，求赞³?