CJOJ 1010【NOIP2003】加分二叉树 / Luogu 1040 加分二叉树 /CodeVS 1090 加分二叉树

CJOJ 1010【NOIP2003】加分二叉树 / Luogu 1040 加分二叉树（树型动态规划）

Description

设 一个 n 个节点的二叉树 tree 的中序遍历为（ 1,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

Http

CJOJ:http://oj.changjun.com.cn/problem/detail/pid/1010
Luogu:https://www.luogu.org/problem/show?pid=1040
CodeVS:http://codevs.cn/problem/1090/

Source

树型动态规划

解决思路

因为题目给出的是中序遍历，所以有任意一棵子树的中序遍历一定是在一段里面的，我么令F[i][j]表示从i到j的最大加分，用Mayuri[i][j]表示i,j能得到最大加分的根节点。

那么我们可以枚举[i,j]之间的一个点k为根节点，得到状态转移方程F[i][j]=max(F[i][k-1]*F[k+1][j]+Value[k])同时更新Mayuri[i][j]，最后的结果就是F[1][n]。

最后只输出，这里我们采用递归调用的方法，在主函数里调用Outp(1,n)，对于Outp(i,j)来说，我们调用Outp(i,Mayuri[i]-1) Outp(Mayuri[i]+1,j)，分别按照前序遍历输出。

需要注意的是各个变量的初始值赋值，具体请看代码。

代码

#include<iostream>#include<cstdio>#include<cstdlib>#include<cstring>#include<algorithm>using namespace std;const int maxsize=100;const int inf=2147483647;int n;long long Node[maxsize];//存放每个点的权值long long F[maxsize][maxsize];//F[i][j]如题解中所示int Mayuri[maxsize][maxsize];//Mayuri[i][j]表示i,j这棵树的根节点void Outp(int l,int r);int main(){    memset(F,0,sizeof(F));    cin>>n;    for (int i=1;i<=n;i++)        cin>>Node[i];    for (int i=1;i<=n;i++)        F[i][i-1]=1;//F[][]的初始化    for (int i=1;i<=n;i++)    {        F[i][i]=Node[i];//F的初始化        Mayuri[i][i]=i;//Mayuri的初始化    }    for (int i=n;i>=1;i--)    {        for (int j=i+1;j<=n;j++)        {            for (int k=i;k<=j;k++)                if (F[i][k-1]*F[k+1][j]+Node[k]>F[i][j])                {                    F[i][j]=F[i][k-1]*F[k+1][j]+Node[k];                    Mayuri[i][j]=k;                    //cout<<"Update : "<<i<<' '<<j<<' '<<k<<' '<<F[i][j]<<endl;                }        }    }    cout<<F[1][n]<<endl;    Outp(1,n);//递归输出前序遍历    cout<<endl;    return 0;}void Outp(int l,int r){    if (l>r)        return;    cout<<Mayuri[l][r]<<' ';//因为是前序遍历，所以先输出根节点    if (l==r)        return;    Outp(l,Mayuri[l][r]-1);//递归调用    Outp(Mayuri[l][r]+1,r);    return;}