luogu1040加分二叉树【DP】【树的遍历】

>Description
　　设一个n个节点的二叉树tree的中序遍历为（l,2,3,…,n），其中数字1,2,3,…,n为节点编号。每个节点都有一个分数（均为正整数），记第j个节点的分数为di，tree及它的每个子树都有一个加分，任一棵子树subtree（也包含tree本身）的加分计算方法如下：
subtree的左子树的加分× subtree的右子树的加分＋subtree的根的分数
叶子的加分就是叶节点本身的分数。不考虑它的空子树。
试求一棵符合中序遍历为（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

---

>解题思路
在洛谷上AC了但是在学校网站上没有A，所以不鸟学校网站。

中序遍历：左根右。
所以可以根据给出的中序遍历做DP，在每段节点中（中序遍历）枚举根，根的左边就为左子树，右边就为右子树，
$f[i][j]$表示 $i$~$j$ 这段中序遍历的节点为一棵树时可以得到的最高分数。

每次更新$f$时顺便记录一下 $i$~$j$ 的最优值是以哪个节点为根得到的，最后就根据记录的根们递归输出前序遍历就好了

---

>代码

#include <iostream>
#include <cstdio>
#include <cstring> 

using namespace std;

int n, a[30], f[30][30], root[30][30];

void output (int l, int r)
{
	if (l > r) return; //不存在就退出
	if (l == r)
	{
		printf ("%d ", l);
		return;
	}
	printf ("%d ", root[l][r]); //先输出根（前序遍历）
	output (l, root[l][r] - 1); //左
	output (root[l][r] + 1, r); //右
}
int main()
{
	scanf ("%d", &n);
	for (int i = 1; i <= n; i++)
	{
		scanf ("%d", &a[i]);
		f[i][i] = a[i]; //记录最高分数
		root[i][i] = i; //记录每段得到最高分数的根
	}
	for (int k = 2; k <= n; k++) //长度
	 for (int i = 1; i <= n - k + 1; i++) //起始位置
	 {
	 	int j = i + k - 1; //终止位置
	 	for (int t = i; t <= j; t++) //枚举根
	 	{
	 		int sum = 0;
	 		bool check = 0;
	 		if (i <= t - 1)
	 		{
	 			sum = f[i][t - 1];
	 			check = 1;
	 		}
	 		if (t + 1 <= j)
	 		{
	 			if (check) sum *= f[t + 1][j];
	 			else sum = f[t + 1][j];
	 		}
	 		sum += a[t]; //计算分数（空子树不用鸟）
	 		if (sum > f[i][j])
	 	    {
			 	f[i][j] = sum;
				root[i][j] = t;
			} //更新数据
	 	}
	 }
	printf ("%d\n", f[1][n]);
	output (1, n); //输出
	return 0;
}