刷题笔记：P1040 [NOIP2003 提高组] 加分二叉树

题意：

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

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

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

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

1. tree 的最高加分。

2. tree的前序遍历。

输入格式

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

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

输出格式

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

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

数据规模与约定

对于全部的测试点，保证 1≤n<30，节点的分数是小于 100 的正整数，答案不超过 4×10^9。

思路：

树本质是图，是一个连通、无环、顶点数为边数+1的图。

一种中序遍历序列可以还原成许多不同结构的二叉树，但是只对应一种结构的图。

中序遍历序列相同，而结构不同的二叉树区别在于“根”的选择，是一个递归的结构，不仅是整棵树的根的选择，还包括子树的根，子树的子树的根……

代码：

1.严格位置依赖的dp

#include<bits/stdc++.h>
using namespace std;
const int MAX = 50;
typedef long long ll;
ll n;
ll f[MAX][MAX], root[MAX][MAX];

void print(ll l, ll r) {
	if (l > r)return;
	printf("%lld ", root[l][r]);
	if (l == r)return;
	print(l, root[l][r] - 1);
	print(root[l][r]+1,r);
}

int main() {
	scanf("%lld", &n);
	for (int i = 1; i <= n; i++)scanf("%lld", &f[i][i]),f[i][i-1]=1, root[i][i] = i;
	for (int len = 1; len < n; ++len) {
		for (int i = 1; i + len <= n; ++i) {
			int j = i + len;
			f[i][j] = f[i + 1][j] + f[i][i];
			root[i][j] = i;
			for (int k = i + 1; k < j; ++k) {
				if (f[i][j] < f[i][k - 1] * f[k + 1][j] + f[k][k]) {
					f[i][j] = f[i][k - 1] * f[k + 1][j] + f[k][k];
					root[i][j] = k;
				}
			}
		}
	}
	cout << f[1][n] << endl;
	print(1, n);
	return 0;
}

2.记忆化搜索

#include<bits/stdc++.h>
using namespace std;
const int MAX = 50;
int n,v[MAX],dp[MAX][MAX],root[MAX][MAX];

int f(int l,int r){
    if(dp[l][r]>0)return dp[l][r];
    if(l==r)return v[l];
    if(r<l)return 1;
    for(int i=l;i<=r;i++){
        int p=f(l,i-1)*f(i+1,r)+dp[i][i];
        if(p>dp[l][r]){
            dp[l][r]=p;root[l][r]=i;
        }
    }
    return dp[l][r];
}

void print(int l,int r){
    if(r<l)return;
    if(l==r){printf("%d ",l);return;}
    printf("%d ",root[l][r]);
    print(l,root[l][r]-1);
    print(root[l][r]+1,r);
}

int main(){
    scanf("%d",&n);
    for(int i=1;i<=n;i++)scanf("%d",&v[i]),dp[i][i]=v[i];
    printf("%d\n",f(1,n));
    print(1,n);
    return 0;
}