洛谷P1040 加分二叉树题解

dp即可

\(f[i][j]\)表示i到j的加分

相当于区间dp了

#include<cstdio>
using namespace std;
int v[50];
int f[55][55];
int root[55][55];
void print(int l,int r)
{
    if(l>r)return ;
    if(l==r)
    {
        printf("%d ",r);
        return ;
    } 
    int tmp=root[l][r];
    printf("%d ",tmp);
    print(l,tmp-1);
    print(tmp+1,r);
}
int main()
{
    int n;
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&v[i]);
        f[i][i]=v[i];
        f[i][i-1]=1;
    }
    for(int i=n-1;i>=1;i--)
    {
        for(int j=i+1;j<=n;j++)
        {
            for(int k=i;k<=j;k++)
            {
                if(f[i][j]<(f[i][k-1]*f[k+1][j]+f[k][k]))//k为根节点
                {
                    f[i][j]=f[i][k-1]*f[k+1][j]+f[k][k];
                    root[i][j]=k;
                }
            }
        }
    }
    printf("%d\n",f[1][n]);
    print(1,n);//输出中序遍历
    return 0; 
}