[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;
}