NOIP 2003 加分二叉树

NKOJ 1044 加分二叉树

问题描述

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

首先回顾一下二叉树的的遍历的相关知识：

1.有关定义:

① 前序遍历：先访问根节点，再访问左子树，最后访问右子树。
② 中序遍历：先访问左子树，再访问根节点，最后访问右子树。
③ 后序遍历：先访问左子树，再访问右子树，最后访问根节点。

2.一些性质：

① 前序遍历中，最先访问的是整棵树的根节点。
② 后序遍历中，最后访问的是整棵树的根节点。
③ 中序遍历中，根的整棵左子树总在根之前被访问，根的整棵右子树总在根之后被访问。
④ 已知前序遍历和后序遍历，不能得出整棵树的结构；只要知道中序遍历和其他两种遍历之一，就能得出唯一的整棵树的结构。

本题主要利用性质③。由性质③可知，选定一个节点作为根之后，左子树一定在它的左边，右子树一定在它的右边，于是就成了一道裸的区间DP。

DP过程中记录下将区间[l,r]作为子树时得分最高的根，根据DP的最优子结构性质，通过递归的方式就能得出前序遍历。

//代码采用的是记忆化搜索的方式
#include<stdio.h>
#define ll long long
int N,sc[50],rt[50][50];
ll f[50][50];

void DP(int l,int r)
{
    if(f[l][r])return;
    if(l==r){f[l][r]=sc[l];rt[l][r]=l;return;}
    if(l>r){f[l][r]=1;return;}//空子树加分为1

    int i;
    for(i=l;i<=r;i++)//依次讨论根节点
    {
        DP(l,i-1);DP(i+1,r);
        if(f[l][r]<f[l][i-1]*f[i+1][r]+sc[i])
        {
            f[l][r]=f[l][i-1]*f[i+1][r]+sc[i];
            rt[l][r]=i;
        }
    }
}

void DFS(int p,int l,int r)//递归求后序遍历
{
    if(l>r)return;
    if(l==r){printf("%d ",l);return;}
    printf("%d ",p);
    DFS(rt[l][p-1],l,p-1);
    DFS(rt[p+1][r],p+1,r);
}

int main()
{
    int i,j,k;

    scanf("%d",&N);
    for(i=1;i<=N;i++)scanf("%d",&sc[i]),f[i][i]=sc[i];

    DP(1,N);
    printf("%lld\n",f[1][N]);
    DFS(rt[1][N],1,N);
}