【NOIP2003提高组】加分二叉树（区间DP+二叉树中序遍历）

传送门：加分二叉树

思路：状态表示：f[l][r]表示所有中序遍历是[l,r]这段序列的加分二叉树的最大值

           状态转移方程：f[l][r]=max(f[l][r],f[l][k-1]+f[k+1][r]+w[k]);

代码：

#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;
typedef long long LL;
const int N=55;
int n;
int f[N][N],g[N][N];
int w[N];
void dfs(int l,int r)
{
    if(l>r) return;
    int k=g[l][r];
    cout<<k<<' ';
    dfs(l,k-1);
    dfs(k+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;
    dfs(1,n);
;    return 0;
}