计蒜客 奇怪的二叉树 （区间DP）

 参考博客

/*
dp[i][j]表示从i到j的中序遍历表示的子树长度
/*/

#include <bits/stdc++.h>

using namespace std;

const int N=35;

int n,a[N],dp[N][N],root[N][N];

void print_way(int l,int r){
    if(l>r) return ;
    printf("%d ",root[l][r]);
    print_way(l,root[l][r]-1);
    print_way(root[l][r]+1,r);
}

int main()
{
    scanf("%d",&n);
    for(int i=1;i<=n;i++){
        scanf("%d",&a[i]);
    }
    for(int i=1;i<=n;i++){
        dp[i][i]=a[i];
        root[i][i]=i;
    }
    for(int l=2;l<=n;l++){
	for(int i=1;i<=n-l+1;i++){
	    int j=i+l-1;
            for(int k=i+1;k<j;k++){
                if(dp[i][k-1]*dp[k+1][j]+a[k]>dp[i][j]){
                    dp[i][j]=dp[i][k-1]*dp[k+1][j]+a[k];
                    root[i][j]=k;
                }
            }
            if(dp[i+1][j]+a[i]>dp[i][j]){
                dp[i][j]=dp[i+1][j]+a[i];
                root[i][j]=i;
            }
            if(dp[i][j-1]+a[j]>dp[i][j]){
                dp[i][j]=dp[i][j-1]+a[j];
                root[i][j]=j;
            }
        }
    }
    printf("%d\n",dp[1][n]);
    print_way(1,n);
    return 0;
}