AcWing479 加分二叉树（区间dp）

链接: link.
这道题我们根据题意，要我们求得二叉树最大的加分，并输出字典序最小的方案。貌似是对二叉树的考察，但并不是，我们需要用区间dp解决这道题。

首先我们看一下什么是区间dp

区间dp

1. 区间dp 是一种建立在线性结构上的对区间的动态规划，求解一段区间上的最优解。
2. 区间dp 我们一般是将大区间转化成小区间，然后对每个小区间求最优解，进而得到大区间的最优解。
3. 区间dp问题和背包问题一样是有模板的，模板有三要素：区间长度，区间起点和终点，分割

下面我们看一下模板

for(int len=1;len<=n;len++)//枚举长度
    {
        for(int l=1;l+len-1<=n;l++)//枚举起点，并保证终点不越界
        {
            int r = l+len-1;//终点
            for(int k=l;k<=r;k++)
                //转移方程，something根据题目进行填写
                dp[l][r] = max(dp[l][r],dp[l][k]+dp[k+1][r]+something);
        }
    }

显然时间复杂度为O(N^3)
模板优化可以看这个博客链接: link.

问题解决

我们回到题目，题目给出二叉树的中序遍历，是线性结构，又让我们求出最大加分，我们可以拆分到若干子树求最大加分，这不就满足了区间dp的条件了吗。对于解决输出先序遍历，我们可以申请一个数组存储我们每次遍历的点。

#include <bits/stdc++.h>

using namespace std;

const int N = 30;
int n;
int w[N];
int dp[N][N],path[N][N];

void print(int l,int r)//先序遍历输出
{
    if(l>r) return ;
    int root = path[l][r];
    cout<<root<<' ';
    print(l,root-1);
    print(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)//区间长度为1时进行特殊处理
            {
                dp[l][r] = w[l];
                path[l][r] = l;
            }
            else
            {
                for(int k=l;k<=r;k++)
                {//对于左右边界进行判断，叶子节点值设为1
                    int left = k==l ? 1:dp[l][k-1];
                    int right = k==r ? 1:dp[k+1][r];
                    int ans = left*right+w[k];
                    if(ans>dp[l][r])
                    {
                        dp[l][r] = ans;
                        path[l][r] = k;
                    }
                }
            }
        }
    }
    cout<<dp[1][n]<<endl;
    print(1,n);
    return 0;
}