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

题目描述

设一个 n 个节点的二叉树 tree 的中序遍历为（1,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 行 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

说明/提示

数据规模与约定

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

思路：给出了中序序列，中序序列是左子树→根节点→右子树。所以可以在一个[l,r]的区间中选取一个断点k将[l,k-1]、k、[k+1,r]合并成[l,r]，是不是与区间dp很相似呢？（但是我先用了递归去写，超时了/(ㄒoㄒ)/~~）。然后输出前序序列，可以再用一个二维数组存储[i,j]的断点。

dp核心状态转移方程：

但是这一题要考虑多种情况，我根据这个思路分成了三个部分（i==j、i>k-1、k+1>j）。

AC代码：

#include<bits/stdc++.h>
#define ll long long
using namespace std;
ll a[60];
ll dp[60][60];
int p[60][60];
int n;
void print(int l,int r){
	if(l>r)return ;
	printf("%d ",p[l][r]);
	print(l,p[l][r]-1);
	print(p[l][r]+1,r);
}
int main(){
	cin>>n;
	for(int i=1;i<=n;i++)
	cin>>a[i];
	for(int len=1;len<=n;len++)
		for(int i=1;i<=n-len+1;i++)
		{
			if(len==1){
				dp[i][i]=a[i];
				p[i][i]=i;
			}else{
				int f;
				for(int k=i;k<=i+len-1;k++)
				{
					if(i>k-1)f=dp[k+1][i+len-1]+a[k];
					else if(k+1>i+len-1)f=dp[i][k-1]+a[k];
					else f=dp[i][k-1]*dp[k+1][i+len-1]+a[k];
					if(f>dp[i][i+len-1]){
						dp[i][i+len-1]=f;
						p[i][i+len-1]=k;
					}
				}
			}
		}
	cout<<dp[1][n]<<endl;
	print(1,n);
	return 0;
}