【SSL_1033】&【洛谷_P1040】加分二叉树

加分二叉树

Time Limit:1000MS
Memory Limit:65536K

Description

设一个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的前序遍历

Input

第1行：一个整数n（n＜30），为节点个数。
第2行：n个用空格隔开的整数，为每个节点的分数（分数＜100）。

Output

第1行：一个整数，为最高加分（结果不会超过4,000,000,000）。
第2行：n个用空格隔开的整数，为该树的前序遍历。

Sample Input

5
5 7 1 2 10

Sample Output

145
3 1 2 4 5

说明/提示

n<30。

分数<100。

【题目来源】

NOIP 2003 提高组第三题

思路

区间DP,$f[i][j]$表示i~j的最大值，
$f[i][j]=max(f[i][j],f[i][k-1]\ast f[k+1][j]+f[k][k])$，$i<=k<j$
再用$root[i][j]$表示 $i$~$j$区间的根，最后递归求出前序遍历。

代码

#include<stdio.h>
#include<iostream>
#include<cstring>
using namespace std;
int n,f[31][31],root[31][31],a[31],ans;
void in()
{
	cin>>n;
	//memset(f,1,sizeof(f));
	for(int i=0; i<=n; i++)
		for(int j=0; j<=n; j++)
			f[i][j]=1;
	for(int i=1; i<=n; i++)
	{
		scanf("%d",&a[i]);
		f[i][i]=a[i];
		root[i][i]=i;
	}
}
void dp()
{
	for(int len=1; len<=n; len++)//枚举区间长度
	{
		for(int i=1; i<=n-len; i++)//枚举起始位置
		{
			int j=i+len;
			for(int k=i; k<j; k++)//枚举根
			{
				if(f[i][j]<(f[i][k-1]*f[k+1][j]+f[k][k]))
				{
					f[i][j]=f[i][k-1]*f[k+1][j]+f[k][k];//状态转移
					root[i][j]=k;
				}
				ans=max(ans,f[i][j]);
			}

		}
	}
}
void solve(int l,int r)
{
	int mid=root[l][r];
	if(mid) cout<<mid<<' ';
	if(root[l][mid-1]) solve(l,mid-1);
	if(root[mid+1][r]) solve(mid+1,r);
}
int main()
{
	in();
	dp();
	cout<<ans<<'\n';
	solve(1,n);
}