P1040 加分二叉树(区间DP)

版权声明:个人博客:www.jingyile.cn 萌新发博文积累经验,欢迎各位大佬指导!!! https://blog.csdn.net/JYL1159131237/article/details/78355042

P1040 加分二叉树 https://www.luogu.org/problemnew/show/1040

题目描述

设一个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行:一个整数n(n<30),为节点个数。

第2行:n个用空格隔开的整数,为每个节点的分数(分数<100)。

输出格式:

第1行:一个整数,为最高加分(结果不会超过4,000,000,000)。

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


算法分析:

结合问题,如果整棵树的权值最大,必然有左子树的权值最大,右子树的权值也最大,符合最优性原理。

而却不是一道树规的题目。因为我们可以用区间动规的模型解决掉:直接定义一个f[i][j]表示从i到j的最大值,则

f[i][j]=max(f[i][k-1]*f[k+1][j]+f[k][k])

枚举k即可。


接下来是如何建树的问题,只有把树建好了,才能输出其前序遍历。于是,我们看到了两个关键词:二叉树,中序遍历。有了这两个关键词,加上区间动规,这棵树就能建起来了。根据二叉树的特性来建树。所以这颗树的前序遍历,只需要边动规边记录下root[i][j]=k表示i到j的根为k即可确定树的构造。


前序遍历(DLR) 
  前序遍历首先访问根结点然后遍历左子树,最后遍历右子树。在遍历左、右子树时,仍然先访问根结点,然后遍历左子树,最后遍历右子树。
  若二叉树为空则结束返回,否则: 
  (1)访问根结点 
  (2)前序遍历左子树 
  (3)前序遍历右子树 
  注意的是:遍历左右子树时仍然采用前序遍历方法。 
  如上图所示二叉树
  前序遍历,也叫先根遍历,遍历的顺序是,根,左子树,右子树
  遍历结果:ABCDEF 
  中序遍历,也叫中根遍历,顺序是 左子树,根,右子树 
  遍历结果:CBDAEF
  后序遍历,也叫后根遍历,遍历顺序,左子树,右子树,根
  遍历结果:CDBFEA

AC代码:

#include<iostream>
#include<memory.h>
using namespace std;
int n;
int f[35][35];
int root[35][35];//root[a][b]表示a,b区间里父节点的编号。
int get_Max(int left,int right)
{
	if(left>right)
	{
		return 1;
	}
	if(f[left][right]==0)
	{
		for(int i=left;i<=right;i++)
		{
			int temp=f[left][right];
			f[left][right]=max((get_Max(left,i-1)*get_Max(i+1,right)+f[i][i]),f[left][right]);
			if(temp<f[left][right])

			{
				root[left][right]=i;//如果找到比当前更大的就更新,同时记录树根节点
			}
		}
	}
	return f[left][right];
}
void dfs(int left,int right)
{
	if((left>right))
	{
		return ;
	}
	else
	{
		cout<<root[left][right]<<' ';
	    dfs(left,root[left][right]-1);
	    dfs(root[left][right]+1,right);
	}
}
int main(){
	cin>>n;
	memset(f,0,sizeof(f));//数组初始化为0
	memset(root,0,sizeof(root));
	for(int i=1;i<=n;i++)
	{
		int m;
		cin>>m;
		f[i][i]=m;
		root[i][i]=i;
	}
	cout<<get_Max(1,n)<<endl;
	dfs(1,n);
	return 0;
}



没有更多推荐了,返回首页