动态规划加分二叉树问题

动态规划加分二叉树问题

题目描述

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

subtree 的左子树的加分 X 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

说明/提示

n<30。

分数 <100。

解题思路

根据得分的计算方法，我们可以知道要想分数最大，就要让左子树和右子树分数最大，而左子树和右子树又需要他们的左子树和右子树分数最大，所以这道题可以用动态规划来做。

用root[i][j]表示从节点i到节点j组成的二叉树的根节点

用f[i][j]表示从节点i到节点j组成的二叉树所能得到的最大分数

那么状态转移方程就是f[i][j]=max(f[i][j],f[i][k-1]*f[k+1][j]+f[k][k])，表示以k为根节点会不会使分数更高

目标状态是f[1][n]

至于输出前序遍历，就是递归不断输出根节点，详见代码

完整代码

#include <iostream>
using namespace std;
int n;
int a[30];
int root[30][30];//root[i][j]表示从节点 i到节点 j这段树的根节点
int f[30][30];//f[i][j]表示从节点 i到节点 j这段树能得到的最大分数 

void out(int l,int r)//递归输出前序遍历 
  {
  	if(l>r) return;//已全部完成 
  	if(l==r) //子树为空 
  	  {
  	  	cout<<l<<" ";
  	  	return;
	  }
	cout<<root[l][r]<<" ";
	out(l,root[l][r]-1);//继续左子树 
	out(root[l][r]+1,r);//继续右子树 
  } 

int dp()//动归找最大分数 
  {
  	for(int i=n;i>=1;i--)//左边界 
  	  {
  	  	for(int j=i+1;j<=n;j++)//右边界 
  	  	   {
  	  	   	 for(int k=i;k<=j;k++)//根节点 
  	  	   	    {
  	  	   	      //f[i][j]=max(f[i][j],f[i][k-1]*f[k+1][j]+f[k][k]);//取分数较大的 
  	  	   	      if(f[i][j]<(f[i][k-1]*f[k+1][j]+f[k][k])) //如果以 k为根节点分数更大，则以 k为根节点 
						 {
						   f[i][j]=f[i][k-1]*f[k+1][j]+f[k][k];
						   root[i][j]=k;	
						 }
				}
		   }
	  }
  } 

int main()
  {
  	cin>>n;
  	for(int i=1;i<=n;i++)
  	  {
  	  	cin>>a[i];
	  }
	for(int i=1;i<=n;i++)
	  {
	  	f[i][i]=a[i];//节点加分赋值 
	  	f[i][i-1]=1;//子树为空，加分为 1 
	  } 
	dp();
	cout<<f[1][n]<<endl;
	out(1,n);
	return 0;
  }