DP学习——树形DP——P1040 [NOIP2003 提高组] 加分二叉树

题目描述

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

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

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

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

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

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

输入格式

第 111 行 111 个整数 nnn，为节点个数。

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

输出格式

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

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

思路

关注这个 左子树*右子树+根 我只要知道左子树分数和右子树分数和根的分数（已给出），不就可以了吗？管他子树长什么样！
所以，我们fff数组存的就是最大分数，怎么存呢？
我们发现：子树是一个或多个节点的集合。
那么我们可不可以开一个f[i][j]f[i][j]f[i][j]来表示节点i到节点j成树的最大加分呢？可以先保留这个想法（毕竟暂时也想不到更好的了）。

如果这样话，我们就来设计状态转移方程。
按照刚刚的设计来说的话，我们的答案就是f[1][n]f[1][n]f[1][n]了，那么我们可以从小的子树开始，也就是len，区间长度。有了区间长度我们就要枚举区间起点，i为区间起点，然后就可以算出区间终点j。
通过加分二叉树的式子我们可以知道，二叉树的分取决于谁是根，于是我们就在区间内枚举根k。
特别的，f[i][i]=a[i]f[i][i]=a[i]f[i][i]=a[i]其中a[i]为第i个节点的分数。
因为是要求最大值，所以我们就可以设计出

f[i][j]=MAX(f[i][k−1]∗f[k+1][j]+f[k][k])f[i][j]=MAX(f[i][k-1]*f[k+1][j]+f[k][k]) f[i][j]=MAX(f[i][k−1]∗f[k+1][j]+f[k][k])

于是乎，我们就自己设计出了一个dp过程，因为是顺着来的，所以很少有不成立的。

至于输出前序遍历，我们再设计一个状态root[i][j]root[i][j]root[i][j]来表示节点i到节点j成树的最大加分所选的根节点。
所以我们按照根−>左−>右根->左->右根−>左−>右的顺序递归输出即可。

#include<bits/stdc++.h>

using namespace std;

const int MAX_SIZE = 40;

int nums[MAX_SIZE];
long long dp[MAX_SIZE][MAX_SIZE];//第i个数到第j个数组成数的最大分数
int root[MAX_SIZE][MAX_SIZE];

bool first = true;

void print(int l, int r)
{
    if(l > r)
        return;
    if(first)
    {
        cout<<root[l][r];
        first = false;
        if(l == r)
            return;
    }
    else
    {
        cout<<' '<<root[l][r];
        if(l == r)
            return;
    }
    print(l, root[l][r] - 1);
    print(root[l][r] + 1, r);
}


int main()
{
    int n;
    cin >> n;

    for(int i = 1; i <= n; i++) {
        cin >> nums[i];
        dp[i][i] = nums[i];
        root[i][i] = i;
    }

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

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