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行：一个整数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

思路：

看到题目，以为是二叉树，竟然要根据中序遍历求前序遍历！（心想这怎么可能）后来经过大佬指点，发现记忆化搜索就可以。

代码：

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>

#define loop( i, a, b )	for( int i = a; i <= b; i++ )

using namespace std;

//root[i][j]为i、j两个节点的根节点，f数组存分数。
int n, v[10010], root[2002][2002], f[2002][2002];

int hahasearch( int l, int r ) {
	if( f[l][r] > 0 )	return f[l][r];
	if( l == r )	return v[l];
	if( l > r )	return 1;	//节点为空时。
	loop( i, l, r ) {
		int p;
		//得到从l到r某个点作为根节点所得到的最多加分。
		p = hahasearch( l, i - 1 ) * hahasearch( i + 1, r ) + f[i][i];
		if( p > f[l][r] ) {
			f[l][r] = p;
			root[l][r] = i;	//顺带存l、r两个子节点的根节点。
		}
	}
	return f[l][r];
}

//求先序遍历
void find_tree( int l, int r ) {
	if( l > r )	return;	//节点为空。
	if( l == r ) {	//找到叶节点。
		printf( "%d ", l );
		return;
	}
	printf( "%d ", root[l][r] );	//输出节点。
	//从根节点向左向右
	find_tree( l, root[l][r] - 1 );
	find_tree( root[l][r] + 1, r );
}

int main() {
	scanf( "%d", &n );
	loop( i, 1, n ) {
		scanf( "%d", &v[i] );
		f[i][i] = v[i];	//初始化f数组
	}
	int ans = hahasearch( 1, n );
	printf( "%d\n", ans );
	find_tree( 1, n );
	return 0;
}