【记忆化搜索】[NOIP2003]加分二叉树 binary

题三    加分二叉树 binary

【问题描述】

    设一个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

    31 2 4 5

这一题给出了中序遍历（即1 , 2 , 3 , 4 , 5 , ... , n），我们就可以枚举当前区间 [ l , r ] 的根 x ，然后就可以得知他的左子树为 [ l , x - 1 ]，右子树为 [ x+1 , r ] 两个区间

这样一直递归下去就可以得到最终的解

那怎样递归呢？用dfs(l,r) 表示当前的区间，每次枚举根 x ，然后依次递归下去即可，然后分别得到左子树和右子树的值，算的当前的值，更新即可

大概代码如下

以下定义 long long 为 LL
score[x] 表示 x 这个节点的分数

LL dfs(int l,int r)//区间为[l,r]
{
	if(l==r) return score[l];//区间大小为1 ，返回当前得分
	if(l>r) return 1;//没有这个子树，返回 1
	LL res=0;
	LL left=1,right=1;
	for(int x=l;x<=r;x++)//枚举当前子树的根 x
	{
		left=dfs(l,x-1);//递归左子树
		right=dfs(x+1,r);//递归右子树
		if(res<left*right+score[x]) res=left*right+score[x];//更新
	}
	return res;
}

主函数只需要调用 res=dfs(1,n) 即可

这样就可以求出最优值了，但是如何求前序遍历呢？

这个简单，再多带入一个为string的参数，来保存当前区间的前序遍历，

dfs参数变为

LL dfs(int l,int r,string &ll)//区间为[l,r]

在更新的时候就可以变为

用number[x]把 x 转化为 string
if(res<left*right+score[x])//更新
{
	res=left*right+score[x];
	ll=number[x]+" "+lans+" "+rans;//前序遍历为：根+左子树+右子树
}

注意上面为什么要带成 string &ll 呢？主要是我们的 dfs 已经返回一个 long long 值了，但是我们要多维护一个，就可以开一个结构体一起返回，但是代码就增多了，并且耗时也大了，所以我们就可以带入引用参数，改变了值就改变了上一层带入时的值，就很方便了

不过还有一个小问题，ll=number[x]+" "+lans+" "+rans;    注意这句话，如果我们的左子树为空，相当于就成了 ll=number[x]+" "+" "+rans;  也就是多了一个空格！！所以输出的时候还需要处理一下

这样我们就能得到一个正确的答案了

那时间呢？

dfs？

O(2^N) ！

肯定要超时！

怎么办呢？

对！记忆化搜索！

加上就可以轻松AC了！

完成情况（RQNOJ测评）

#include<cstdio>
#include<algorithm>
#include<string>
#include<iostream>
#include<cstring>
using namespace std;
typedef long long LL;
const int N=30+10;
int vis[N][N],f[N][N];
string ss[N][N];
const string number[]=//用number[x]把 x 转为string类
{"0","1","2","3","4","5","6","7","8","9","10",
"11","12","13","14","15","16","17","18","19","20","21","22","23","24","25",
"26","27","28","29","30","31","32","33","34","35","36","37","38","39","40"};
int n;
int a[N],score[N];
string order="";

LL dfs(int l,int r,string &ll)//区间为[l,r]
{
	if(l==r) {ll=number[l];return score[l];}//区间大小为1
	if(l>r) return 1;
	if(vis[l][r]) {ll=ss[l][r];return f[l][r];}
	vis[l][r]=true;//记忆化标记
	LL res=0;
	LL left=1,right=1;
	for(int x=l;x<=r;x++)//枚举当前子树的根 x
	{
		string lans="",rans="";
		left=dfs(l,x-1,lans);//递归左子树
		right=dfs(x+1,r,rans);//递归右子树
		if(res<left*right+score[x])
		{
            f[l][r]=res=left*right+score[x];
			ss[l][r]=ll=number[x]+" "+lans+" "+rans;
		}
	}
	return res;
}

int main()
{
	freopen("binary.in","r",stdin);
	freopen("binary.out","w",stdout);
	scanf("%d",&n);
	for(int i=1;i<=n;i++) scanf("%d",&score[i]);
	LL res=dfs(1,n,order);
	cout<<res<<endl;
	for(int i=0;i<order.length();i++)//处理多余空格。。。写得不好，有点麻烦。。。
	{
		if(order[i]==' '&&i>0&&order[i-1]==' ') continue;
		else if(order[i]==' '&&i>0&&order[i-1]!=' ') cout<<' ';
		else cout<<order[i];
	}
	return 0;
}