加分二叉树 vijos P1100

题目链接： 点我

  加分二叉树

描述

设一个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个用空格隔开的整数，为该树的前序遍历。

样例输入1[复制]

5
5 7 1 2 10

样例输出1[复制]

145
3 1 2 4 5

         思路：枚举每个区间的根，取最大加分并记录相应的根。dp[i][j]=max(dp[i][j],dp[i][k-1]*dp[k+1][j]+dp[k][k]);dp[i][j]代表区间【i,j]的最大加分，用d[i][j]记录相应的根。

          记忆化代码：

#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
using namespace std;
const int maxn=1e5+100;
const int MOD=1e9+7;
typedef long long LL;
typedef unsigned long long llu;
LL dp[40][40];
int d[40][40];
 void dfs(int a,int b)
{
	if(a>=b) return;
	int i,k=a;
	LL ans=0;
	for(i=a;i<=b;i++)
	{
		LL tmp=1;
		if(i-1>=a)
		{
			if(dp[a][i-1]==0)
			dfs(a,i-1);
			tmp*=dp[a][i-1];
		}
		if(i+1<=b)
		{
			if(dp[i+1][b]==0) dfs(i+1,b);
			tmp*=dp[i+1][b];
		}
		tmp+=dp[i][i];
		if(ans<tmp) ans=tmp,k=i;
	}
	dp[a][b]=ans;
	d[a][b]=k;
}
void print(int a,int b)
{
	if(a>b) return ;
	printf("%d ",d[a][b]);
	if(d[a][b]>a) print(a,d[a][b]-1);
	if(d[a][b]<b) print(d[a][b]+1,b);
}
int main()
{
	//freopen("1.out","w",stdout);
	int n,i;
	scanf("%d",&n);
	memset(dp,0,sizeof dp);
	for(i=1;i<=n;i++)
	{
		scanf("%I64d",&dp[i][i]);
		d[i][i]=i;
	}
	dfs(1,n);
	printf("%I64d\n",dp[1][n]);
	
	print(1,n);
	return 0;
}

      另一种代码：

#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
using namespace std;
const int maxn=1e5+100;
const int MOD=1e9+7;
typedef long long LL;
typedef unsigned long long llu;
LL dp[40][40];
int d[40][40];
void print(int a,int b)
{
	if(a>b) return ;
	printf("%d ",d[a][b]);
	if(d[a][b]>a) print(a,d[a][b]-1);
	if(d[a][b]<b) print(d[a][b]+1,b);
}
int main()
{
	int n;
	int i,j,k,v;
	memset(dp,0,sizeof dp);
	scanf("%d",&n);
	for(i=1;i<=n;i++)
	{
		scanf("%I64d",&dp[i][i]);
		d[i][i]=i;
	}
	for(i=1;i<n;i++)
	{
		for(j=1;i+j<=n;j++)
		{
			k=i+j;
			dp[j][k]=dp[j][j]+dp[j+1][k];
			d[j][k]=j;
			if(dp[j][k]<dp[j][k-1]+dp[k][k])
			{
				dp[j][k]=dp[j][k-1]+dp[k][k];
				d[j][k]=k;
			}
			for(v=j+1;v<k;v++)
			{
				if(dp[j][k]<dp[j][v-1]*dp[v+1][k]+dp[v][v])
				{
					dp[j][k]=dp[j][v-1]*dp[v+1][k]+dp[v][v];
					d[j][k]=v;
				}
			}
		}
	}
	printf("%I64d\n",dp[1][n]);
	print(1,n);
	return 0;
}