[NOIP2003 提高组] 加分二叉树题解

Shen_Ji_

已于 2024-06-09 13:43:13 修改

阅读量908

点赞数 30

分类专栏：树文章标签：算法动态规划 c++

于 2024-06-09 13:42:10 首次发布

本文链接：https://blog.csdn.net/Shen_Ji_/article/details/139546249

版权

树专栏收录该内容

2 篇文章 0 订阅

订阅专栏

题目

洛谷题目

题目描述

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

$\text{subtree}$ 的左子树的加分 $\times$ $\text{subtree}$ 的右子树的加分 $+$ $\text{subtree}$ 的根的分数。

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

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

$\text{tree}$ 的最高加分。
$\text{tree}$ 的前序遍历。

输入格式

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

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

输出格式

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

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

样例 #1

样例输入 #1

5
5 7 1 2 10

样例输出 #1

145
3 1 2 4 5

提示

数据规模与约定

对于全部的测试点，保证 $\leq n< 30$ ，节点的分数是小于 $100$ 的正整数，答案不超过 $\times 10^9$ 。

题目分析

考虑到本题数据范围，用暴力枚举的方法显然会超时。
再对题目进行分析，能发现每一个点都是由其子树的得分运算得出，所以我们可以将求加分及子树的问题拆解为若干个子问题。即从小节点向大节点进行运算，有没有感觉有一点眼熟？没错，其实就是一个区间 $d p$ ¹

根据题目描述，可得到以下状态转移方程式：

dp[x][y][0]=max(dp[x][y][0],a[z]+dp[x][z-1][0] $*$ dp[z+1][y][0])

具体做法

写dp数组

int dp[100][100][4];//0代表区间最大值，1代表根节点位置，2代表左区间根节点，3代表右区间根节点

从短到长dp

for(int len=1;len<=n;len++)//长度从短到长
		for(int i=1;i+len-1<=n;i++)
		{
			int j=i+len-1;
			if(len==1) dp[i][j][0]=a[i];
			else for(int k=i;k<=j;k++)
			{
				dp[i][j][0]=f(i,j,k);
			}
		}

计算当前节点

int f(int x,int y,int z)
{
	if(dp[x][y][0]<a[z]+dp[x][z-1][0]*dp[z+1][y][0])
	{
		if(x==1&&y==n) q=z;
		dp[x][y][1]=z;
		dp[x][y][2]=dp[x][z-1][1];
		dp[x][y][3]=dp[z+1][y][1];
	}
	return max(dp[x][y][0],a[z]+dp[x][z-1][0]*dp[z+1][y][0]);
}

完整代码

#include <iostream>
#include <cstring>
using namespace std;
long long n,dp[100][100][4],a[100],b[100][3],q;
int f(int x,int y,int z)
{
	if(dp[x][y][0]<a[z]+dp[x][z-1][0]*dp[z+1][y][0])
	{
		if(x==1&&y==n) q=z;
		dp[x][y][1]=z;
		dp[x][y][2]=dp[x][z-1][1];
		dp[x][y][3]=dp[z+1][y][1];
	}
	return max(dp[x][y][0],a[z]+dp[x][z-1][0]*dp[z+1][y][0]);
}
void ot(int x,int y,int q)
{
	if(q==0) return ;
	cout<<q<<" ";
	ot(x,q-1,dp[x][y][2]);
	ot(q+1,y,dp[x][y][3]);
}
int main()
{
	memset(dp,0,sizeof(dp));
	cin>>n;
	for(int i=1;i<=n*2;i++)
		for(int j=1;j<=n*2;j++)
			dp[i][j][0]=1;
	for(int i=1;i<=n;i++)
	{
		cin>>a[i];
		dp[i][i][1]=i;
	}
	for(int len=1;len<=n;len++)
		for(int i=1;i+len-1<=n;i++)
		{
			int j=i+len-1;
			if(len==1) dp[i][j][0]=a[i];
			else for(int k=i;k<=j;k++)
			{
				dp[i][j][0]=f(i,j,k);
			}
		}
	cout<<dp[1][n][0]<<endl;
	ot(1,n,q);
	return 0;
}