DP - 区间DP(求具体方案) - NOIP 2003 - 加分二叉树

DP - 区间DP - NOIP 2003 - 加分二叉树

设一个n个节点的二叉树tree的中序遍历为（1,2,3,…,n），其中数字1,2,3,…,n为节点编号。

每个节点都有一个分数（均为正整数），记第i个节点的分数为di，tree及它的每个子树都有一个加分，任一棵子树subtree（也包含tree本身）的加分计算方法如下：

subtree的左子树的加分 × subtree的右子树的加分 ＋ subtree的根的分数

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

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

要求输出：

（1）tree的最高加分

（2）tree的前序遍历

输入格式
第1行：一个整数n，为节点个数。

第2行：n个用空格隔开的整数，为每个节点的分数（0<分数<100）。

输出格式
第1行：一个整数，为最高加分（结果不会超过4,000,000,000）。

第2行：n个用空格隔开的整数，为该树的前序遍历。如果存在多种方案，则输出字典序最小的方案。

数据范围
n<30

输入样例：
5
5 7 1 2 10
输出样例：
145
3 1 2 4 5

---

分析：

$二叉树的左右子树的收益的计算是相互独立的，因此可以考虑区间DP。$

$由于划分为左右子树来计算收益，因此可以枚举根节点为分界点。$

$状态表示：f[i][j]:区间[i,j]内的元素构成的二叉树获得的最大收益。$

$$\begin{gathered} 状态计算： \\ 区间[l,r]，根节点为k：f[l][r]=max(f[l][k-1]\times f[k+1][r]+w[k]，k\in [l,r]。 \end{gathered}$$

$本题需要记录具体方案，另开数组g存储区间[l,r]的根节点k，即g[l][r]=k。$

注意：

$当根节点为区间端点时(k=l或k=r)，仅存在一颗子树，此时将另一颗子树赋值为1再计算。$

$另外，前序遍历可以采用递归输出。顺序为根->左->右。$

代码：

#include<iostream>
#include<cstring>
#include<algorithm>

using namespace std;

const int N=35;

int n,w[N],f[N][N],g[N][N];

void pre(int l,int r)
{
    if(l>r) return ;
    int root = g[l][r];
    cout<<root<<' ';
    pre(l,root-1);
    pre(root+1,r);
    
}

int main()
{
    cin>>n;
    for(int i=1;i<=n;i++) cin>>w[i];
    
    for(int len=1;len<=n;len++)
        for(int l=1;l+len-1<=n;l++)
        {
            int r=l+len-1;
            if(len==1)
            {
                f[l][r]=w[l];
                g[l][r]=l;
            }
            else
            {
                for(int k=l;k<=r;k++)
                {
                    int left = k==l ? 1 : f[l][k-1];
                    int right = k==r ? 1 : f[k+1][r];
                    int score=left*right+w[k];
                    if(f[l][r]<score)
                    {
                        f[l][r]=score;
                        g[l][r]=k;
                    }
                }
            }
        }
        
    cout<<f[1][n]<<endl;
    pre(1,n);
    
    return 0;
}