加分二叉树

#include<iostream>
#include<algorithm>
#include<cstring>
#include<string>
#include<cstdio>
#include<cstdlib>
#include<cmath>
using namespace std;
inline int rd()
{
    int x=0,f=1;
    char ch=getchar();
    for(;!isdigit(ch);ch=getchar()) if(ch==‘-‘) f=-1;
    for(;isdigit(ch);ch=getchar()) x=x*10+ch-‘0‘;
    return x*f;
}
inline void write(int x)
{
    if(x<0) putchar(‘-‘),x=-x;
    if(x>9) write(x/10);
    putchar(x%10+‘0‘);
    return ;
}
int s[100006],f[1006][1006];
int set[1006][1006];
void dfs(int l,int r)
{
    if(l>r) return ;
    printf(" %d",set[l][r]);
    dfs(l,set[l][r]-1);
    dfs(set[l][r]+1,r);
}
int main()
{
    int n;
    n=rd();
    for(int i=1;i<=n;i++)
    {
        f[i][i]=rd();//输入可以简化，但是不必要 
        f[i][i-1]=1;//读题一定要仔细，非空子树加分为1 （调了半天） 
        set[i][i]=i;//每个节点分割点为自己 
    }
    /*
       划重点！！！
       这道题乍一眼看上去是树形DP
       但实际上多读几遍题就会发现这是区间DP裸题
       “记 subtree 的左子树加分为 l，右子树加分为 r，subtree 的根的分数为 a，则 subtree 的加分为：l×r+a” 
       这句话生动形象的刻画了这道题区间DP的精髓所在 
       由此推出转移方程
       dp[i][j]=max(dp[i][j],dp[i][k-1]+dp[k+1][j]+dis[k]);
       dp[i][j]表示从节点i到节点j的最大加分
       特别的是因为是树结构，所以k-1和k+1，不能包含根节点 
    */
    for(int l=1;l<=n;l++)
    {
        for(int i=1;i<=n-l;i++)
        {
            int j=i+l;
            for(int k=i;k<=j;k++)
            {
                if(f[i][j]<f[i][k-1]*f[k+1][j]+f[k][k])
                {
                    f[i][j]=f[i][k-1]*f[k+1][j]+f[k][k];
                    set[i][j]=k;//存根节点 
                }
            }
        }
    }
    printf("%d\n",f[1][n]);
    printf("%d",set[1][n]);//输出，因为是先序遍历，所以要用到dfs,注意输出格式有没有某位的空格 
    dfs(1,set[1][n]-1);
    dfs(set[1][n]+1,n);
    return 0;
}