noip2003 加分二叉树 dp

#include<bits/stdc++.h>
#define LL long long
using namespace std;
LL dp[35][35][35];
LL a[35];
int ch[35][35][35][2];
LL dfs(int x,int l,int r)
{
if(dp[x][l][r])
return dp[x][l][r];
LL L=0,R=0;
if(l==x)L=1;
else
{
for(int i=l;i<x;i++)
{
LL p=dfs(i,l,x-1);
if(L<p)
{L=p;ch[x][l][r][0]=i;}
}
}
if(r==x) R=1;
else
{
for(int i=x+1;i<=r;i++)
{
LL p=dfs(i,x+1,r);
if(R<p)
{R=p;ch[x][l][r][1]=i;}
}
}
dp[x][l][r]=L*R+a[x];
if(l==r) dp[x][l][r]--;
return dp[x][l][r];
}
int root;
void dfs2(int x,int l,int r)
{
if(!x) return ;
printf("%d ",x);
dfs2(ch[x][l][r][0],l,x-1);
dfs2(ch[x][l][r][1],x+1,r);
}
int main()
{
int n;scanf("%d",&n);
for(int i=1;i<=n;i++)
scanf("%lld",&a[i]);
LL ans=0;
for(int i=1;i<=n;i++)
{
LL p=dfs(i,1,n);
if(ans<p)
{
ans=p;
root=i;
}
}
printf("%d\n",ans);
dfs2(root,1,n);
return 0;
}