【洛谷 1040】加分二叉树

第一次看到这道题是在一个冬立营的论文上。学习了splay后会对二叉树有进一步的理解，就是无论你怎么提，树的形态只和根节点有关so，变成区间dp了，f[i][j]表示ij区间内的最优解枚举根节点，至于最后的答案嘛，dp的时候储存一下就好了

#include
#include 
#include
using namespace std;
int f[35][35],ans[35][35],ch[35][5];
int n,a[35];

void dfs1(int l,int r){
	if(l>r)return ;
	int u=ans[l][r];
	printf("%d ",u);
	if(l==r)return;
	dfs1(l,u-1);
	dfs1(u+1,r);
}


int main(){
	scanf("%d",&n);
	for(int i=1;i<=n;i++)scanf("%d",a+i),f[i][i]=a[i],ans[i][i]=i;
	for(int len=2;len<=n;len++){
		for(int i=1;i<=n-len+1;i++){
			int j=i+len-1;
			for(int k=i;k<=j;k++){
				int l=f[i][k-1]==0?1:f[i][k-1];
				int r=f[k+1][j]==0?1:f[k+1][j];
				int t=l*r+a[k];
				if(t>f[i][j]){
					f[i][j]=t;ans[i][j]=k;
				}
			}
		}
	}
	printf("%d\n",f[1][n]);
	dfs1(1,n);
	return 0;
}