【Luogu_P1040】加分二叉树

原题链接

问题描述

设一个n个节点的二叉树tree的中序遍历为（l,2,3,…,n），其中数字1,2,3,…,n为节点编号。每个节点都有一个分数（均为正整数），记第i个节点的分数为di，tree及它的每个子树都有一个加分，任一棵子树subtree（也包含tree本身）的加分计算方法如下： subtree的左子树的加分× subtree的右子树的加分＋subtree的根的分数 若某个子树为空，规定其加分为1，叶子的加分就是叶节点本身的分数。不考虑它的空子树。 试求一棵符合中序遍历为（1,2,3,…,n）且加分最高的二叉树tree。
要求输出；
（1）tree的最高加分
（2）tree的前序遍历

【输入格式】

第1行：一个整数n（n＜30），为节点个数。 第2行：n个用空格隔开的整数，为每个节点的分数（分数＜100）。

【输出格式】

第1行：一个整数，为最高加分（结果不会超过4,000,000,000）。 第2行：n个用空格隔开的整数，为该树的前序遍历。

Sample Input

 5
 5 7 1 2 10 

Sample Output

145
3 1 2 4 5

思路：
我们首先分析它是一道什么题，发现如果权值最大，也会有左右子树权值最大，所以满足DP条件。
而很多人会认为这是一道树形DP（好像真的可以用），其实我们只要套区间DP的模板就行：
$$f[i][j]=max(f[i][k-1]\ast f[k+1][j]+a[k])$$
其实最烦的就是求先序遍历，但仔细一想，发现在做DP时，我们就已经以最大加分的树把当前区间的父节点记录了下来，在输出时做一个递归就可以求先序遍历。
不懂遍历的可以参考二叉树遍历

代码：

#include<iostream>
#include<cstdio>
#include<cmath>
#include<algorithm>
#include<queue>
#include<stack>
using namespace std;

long long n;
long long a[1001]; 
long long f[1011][1110], root[1100][1100];

void VL_getfirst(long long x, long long y)
{
   if(root[x][y]!=0)
   	printf("%lld ", root[x][y]);
   if(root[x][root[x][y]-1]!=0)VL_getfirst(x, root[x][y]-1);
   if(root[root[x][y]+1][y]!=0)VL_getfirst(root[x][y]+1, y);
}

int main(){
   scanf("%lld", &n);
   for(long long i=0; i<=n; i++)
   	for(long long j=0; j<=n; j++)
   		f[i][j]=1;
   for(long long i=1; i<=n; i++)
   {
   	scanf("%lld", &a[i]);
   	f[i][i]=a[i];
   	root[i][i]=i;
   }
   for(long long len=1; len<=n; len++)
   {
   	for(long long i=1; i<=n; i++)
   	{
   		long long j=i+len, ans=-999999999;
   		if(j<=n)
   		{
   			for(long long k=i; k<=j; k++)
   			{  
   				if(f[i][k-1]*f[k+1][j]+a[k]>ans)
   				{
   					ans=f[i][k-1]*f[k+1][j]+a[k];
   					root[i][j]=k;
   				}
   			}
   			f[i][j]=ans;
   		} 
   	}
   }
   printf("%lld\n", f[1][n]);
   VL_getfirst(1, n);
   return 0;
}