[NOIP2003]加分二叉树

[NOIP2003]加分二叉树 (nowcoder.com)

题目描述

     设一个n个节点的二叉树tree的中序遍历为（l,2,3,…,n），其中数字1,2,3,…,n为节点编号。每个节点都有一个分数（均为正整数），记第j个节点的分数为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个用空格隔开的整数，为该树的前序遍历。

输入

5
5 7 1 2 10

输出

145
3 1 2 4 5

分析

思路是，对于以rt为根的子树，它的最大分值=左子树最大分值*右子树最大分值+v[rt]

所以搜索枚举左右子树的根，取最大值，更新rt[l][r]（区间[l,r]的根，用于后面输出先序遍历）

注意边界判断：

if(l>r) return 1;

if(l==r) return v[l];

可以使用记忆化剪枝优化时间复杂度

/*
中序遍历：左根右
5
5 7 1 2 10
前序遍历：根左右
3 1 2 4 5

	3 
1		4
  2	      5

若某个子树为空，规定其加分为 1

*/

#include<bits/stdc++.h>
#define LL long long
using namespace std;
inline LL R(){LL a=0,b=1;char c=getchar();while(c<'0'||c>'9'){if(c=='-')b=-1;c=getchar();}while(c>='0'&&c<='9'){a=a*10+c-'0';c=getchar();}return a*b;}
LL n,v[50],rt[50][50],f[50][50];
LL dfs(LL l,LL r){
	if(l>r)return 1;
	if(f[l][r])return f[l][r]; 
	//一是为了记忆化剪枝，二是防止叶子节点在第28行的位置会错误地加 1，因为叶子节点的加分应该是v[i]但28行左右子树都为空，会返回1，然后就额外地多加了一个1 
	LL ma=0;
	for(LL i=l;i<=r;++i){
		LL t=dfs(l,i-1)*dfs(i+1,r)+v[i];
		if(t>ma){
			ma=t;
			rt[l][r]=i;
		}
	}
	return f[l][r]=ma;
}
void dfs2(LL l,LL r){
	if(l>r)return;
	LL x=rt[l][r];
	printf("%lld ",x);
	dfs2(l,x-1);
	dfs2(x+1,r);
}
int main(){
	n=R();
	for(LL i=1;i<=n;++i){
		v[i]=R();
		rt[i][i]=i;
		f[i][i]=v[i];
	}
	LL ans=dfs(1,n);
	printf("%lld\n",ans);
	dfs2(1,n);
	return 0;
}