加分二叉树——记忆化搜索

# [NOIP2003 提高组] 加分二叉树

## 题目描述

设一个 n 个节点的二叉树 tree 的中序遍历为1,2,3,,,n，其中数字 1,2,3,,n 为节点编号。每个节点都有一个分数（均为正整数），记第 i 个节点的分数为 di，tree 及它的每个子树都有一个加分，任一棵子树 subtree（也包含 tree 本身）的加分计算方法如下：

subtree 的左子树的加分 x subtree 的右子树的加分 + subtree 的根的分数。

若某个子树为空，规定其加分为 1，叶子的加分就是叶节点本身的分数。不考虑它的空子树。

试求一棵符合中序遍历为 1,2,3,,n 且加分最高的二叉树 tree。要求输出

1. tree 的最高加分。

2. tree 的前序遍历。

输入格式

第 11行 1 个整数 n，为节点个数。

第 2 行 n 个用空格隔开的整数，为每个节点的分数

输出格式

第 1 行 1 个整数，为最高加分（Ans≤4,000,000,000）。

第 2 行 n 个用空格隔开的整数，为该树的前序遍历。

输入输出样例

输入 #1

5
5 7 1 2 10

输出 #1

145
3 1 2 4 5

解决方法+代码：

#include<iostream>
#include<string.h>
#include<algorithm>
using namespace std;

const int N = 40;

int f[N][N];	//f[i][j]表示从i到j区域的最大加分 
int n;
int a[N];
int root[N][N];		//root[i][j]记录从i到j区域的根节点是多少 

void print(int l,int r) //输出l,r区间的根节点 
{
	if(l>r) return;
	cout<<root[l][r]<<" ";
	print(l,root[l][r]-1);
	print(root[l][r]+1,r);
}

int dfs(int l,int r) //返回l,r区间的最大值 
{
	
	if(l>r) return 1;	//空节点
	
	if(f[l][r]) return f[l][r];
	
	for(int i=l;i<=r;i++)		//让l,r区间的每一个数都为根节点，寻找最大值 
	{
		long long num = dfs(l,i-1) * dfs(i+1,r) + a[i];
		if(f[l][r]<num)
		{
			f[l][r] = num;
			root[l][r] = i;
		}
	}
	return f[l][r];
}

int main()
{
	cin>>n;
	
	for(int i=1;i<=n;i++)
	{
		cin>>a[i];
		f[i][i] = a[i];			//初始化 
		root[i][i] = i;
	}
	
	cout<<dfs(1,n)<<endl;
	
	print(1,n);
	
	return 0;
}