[NOIP2003]加分二叉树
时间限制: 1 Sec 内存限制: 64 MB[提交][状态][我的提交]
题目描述
设一个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
一道DP,模型是石子归并。
Dp[i][j] = max(Dp[i][k - 1] + Dp[k + 1][j] + Poi[k]);
DP孱弱者, 上记忆化搜索,好久不写了。。。
#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<algorithm>
using namespace std;
int Dp[32][32], Poi[32], Pre[32][32];
int tot;
int search(int l ,int r){
if(r < l) return 1;
if(Dp[l][r]) return Dp[l][r];
if(r - l >= 2){
for(int i = l + 1; i < r; i++){
int now = search(l, i - 1) * search(i + 1, r) + Poi[i];
if(Dp[l][r] < now){
Dp[l][r] = now;
Pre[l][r] = i;
}
}
}
else{
Dp[l][r] = Poi[l] + Poi[r];
Pre[l][r] = l;
}
return Dp[l][r];
}
void pre_find(int l, int r){
if(l == r){
printf(" %d", l);
return ;
}
if(!Pre[l][r]) return ;
else printf(" %d", Pre[l][r]);
if(Pre[l][r] - 1 == l)
pre_find(l, l);
else pre_find(l, Pre[l][r] - 1);
if(Pre[l][r] + 1 == r)
pre_find(r, r);
else pre_find(Pre[l][r] + 1, r);
}
int main(){
int N;
scanf("%d", &N);
for(int i = 1; i <= N; i++){
scanf("%d", &Poi[i]);
Dp[i][i] = Poi[i];
}
search(1, N);
printf("%d\n", Dp[1][N]);
printf("%d", Pre[1][N]);
pre_find(1, Pre[1][N] - 1);
pre_find(Pre[1][N] + 1, N);
putchar(10);
}