【洛谷P1040】【ssl1033】加分二叉树【树形记忆化搜索】

Description

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

Input

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

Output

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

Sample Input

5
5 7 1 2 10

Sample Output

145
3 1 2 4 5

分析

这题SSL题库中炸掉了，洛谷才有评测。
输入完直接进入记忆化搜索，先来几个判断是否只有一个节点还是没有子节点。

然后遍历整棵树，每次遍历到一个点都要以这个点为根节点递归它的左子树和右子树**（注意不包括i点本身）**，乘起来，同时加上根节点本身的权值。
后面需要用一个root数组记录点，方便后面输出。

最后要递归输出前序遍历，只需要遍历一遍树（也就是root数组），每次输出根节点，然后递归左右子树，直到结束。

上代码

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
using namespace std;
int n,a[100],f[50][50],root[50][50];
void putpre(int l,int r)//根，左，右 
{
	if(l>r) return;
	cout<<root[l][r]<<' ';//输出当前根 
	putpre(l,root[l][r]-1);//递归输出左子树 
	putpre(root[l][r]+1,r);//递归输出右子树 
}
int dfs(int l,int r)
{
	if(l>r) return 1;//无儿子返回自身（也就是1） 
	if(l==r) 
	{
		root[l][r]=r;//那棵树只有一个节点，返回自身 
		return a[r];
	}
	if(f[l][r]>0)//记忆化 
	{
		return f[l][r];
	}
	int ans=0,t;
	for(int i=l;i<=r;i++)//遍历整棵树 
	{
		int s=dfs(l,i-1)*dfs(i+1,r)+a[i];//按照题意计算，a[i]为根节点本身，分别递归左子树右子树 
		if(s>ans)//打擂台更新答案和存序号 
		{
			ans=s;
			t=i;
		}
	}	
	root[l][r]=t;//保存当前节点的序号 
	return f[l][r]=ans;
} 
int main()
{
    cin>>n;
    for(int i=1;i<=n;i++)
    {
        cin>>a[i];
    }
    cout<<dfs(1,n)<<endl;
    putpre(1,n);
	return 0;
}