• P1040 [NOIP2003 提高组] 加分二叉树


    题目来源

    [NOIP2003 提高组] 加分二叉树 - 洛谷

    题目考点

    动态规划,dp   递归   枚举,暴力   Special Judge   O2优化

    题目描述

    设一个 n 个节点的二叉树 tree 的中序遍历为(1,2,3,…,n),其中数字 1,2,3,…,n 为节点编号。每个节点都有一个分数(均为正整数),记第 ii 个节点的分数为 di​,tree 及它的每个子树都有一个加分,任一棵子树 subtree(也包含 tree 本身)的加分计算方法如下:

    subtree 的左子树的加分 subtree 的右子树的加分 subtree 的根的分数。

    若某个子树为空,规定其加分为 1,叶子的加分就是叶节点本身的分数。不考虑它的空子树。

    试求一棵符合中序遍历为 (1,2,3,…,n) 且加分最高的二叉树 tree。要求输出

    1. tree 的最高加分。

    2. tree 的前序遍历。

    输入格式

    第 1 行 1 个整数 n,为节点个数。

    第 2 行 n 个用空格隔开的整数,为每个节点的分数

    输出格式

    第 1 行 1 个整数,为最高加分(Ans≤4,000,000,000)。

    第 2 行 n 个用空格隔开的整数,为该树的前序遍历。

    输入输出样例

    输入 #1

    5
    5 7 1 2 10
    

    输出 #1

    145
    3 1 2 4 5
    

    说明/提示

    数据规模与约定

    对于全部的测试点,保证 1≤n<30,节点的分数是小于 100 的正整数,答案不超过 4×109。

    题解

    这题是一道比较简单的树形DP。我们可以分成3个任务分别解决,如下:(包含CPP及Pascal代码片段)

    【任务一】采用动态规划方法计算最大分值

    本题可以采用动态规划方法来解决,具体如下:

    设f[i, j]为顶点i . . 顶点j所组成的子树的最大分值。若f[i, j] = -1,则表明最大分值尚未计算出。

    f(i,j)={1 (i>j) ; 顶点i的分数 (i=j) ; max(f{i,k-1}*f{k+1,j}+顶点i的分数 (i

    root[i, j]——顶点i..顶点j所组成的子树达到最大分值时的根编号。当i = j时,root[i, i] := i。

    由于问题没有明显的阶段特征,而是呈现为非线性的树形结构,因此,我们采用后序遍历的顺序来计算状态转移方程。计算过程如下:

    【参考程序片段】{cpp版}

    1. long long search(int L, int r) // 递归计算f[L][r]
    2. {
    3. int k;
    4. long long now, ans; // 当前分值
    5. if (L > r) return 1;
    6. if (f[L][r]== -1) // 若尚未计算出顶点L..顶点r对应子树的最高分值
    7. for(k=L; k<=r; k++) { // 穷举每一个可能的子根k
    8. now = search(L, k-1) * search(k+1, r) + f[k][k];
    9. // 计算以k为根的子树的分值
    10. if(now > f[L][r]) {
    11. //若该分值为目前最高,则记入状态转移方程,并记下子根}
    12. f[L][r] = now;
    13. root[L][r] = k;
    14. }
    15. }
    16. return f[L][r]; {返回顶点L..顶点r对应子树的最高分值}
    17. }

    【参考程序】{Pascal版} function search(L, r: integer) : int64; {递归计算f[L. r]}

    1. var
    2. i: integer;
    3. now: int64; {当前分值}
    4. begin
    5. if L > r then search := 1
    6. else
    7. begin
    8. if f[L, r] = -1 then {若尚未计算出顶点L..顶点r对应子树的最高分值}
    9. for i := L to r do {穷举每一个可能的子根i}
    10. begin
    11. now := search(L, i-1) * search(i+1, r) + f[i, i]; {计算以i为根的子树的分值}
    12. if now > f[L, r] then {若该分值为目前最高,则记入状态转移方程,并记下子根}
    13. begin
    14. f[L, r] := now; way[L, r] := i;
    15. end;
    16. end; {for}
    17. search := f[L, r]; {返回顶点L..顶点r对应子树的最高分值}
    18. end; {else}
    19. end; {search}

    显然,主程序可以通过递归调用search(1, n)来计算最高分值。算法的时间复杂度为O(n^2)。

    【任务二】输出加分二叉树的前序遍历

    递归调用search(1, n)后得出的way给出了加分二叉树的结构,其中way[i, j]为该树中顶点i..顶点j的根序号。由于二叉树中序遍历的顺序为1..n,因此,1..way[i, j]-1为左子树,way[i, j] + 1 .. j为右子树。现按照根→左子树→右子树的顺序对加分二叉树进行前序遍历。

    注意:由于数字之间用空格隔开,而第一个数字前没有空格,因此,设firstwrite为首数字标志。一旦输出了第一个数字,firstwrite设为false,表明在输入新的数字前需要加空格。

    由此,得出前序遍历加分二叉树的算法:

    【参考程序片段】{CPP版}

    1. // 前序遍历顶点L..顶点r对应的子树
    2. void preorder(int L, int r)
    3. {
    4. if (L > r) return;
    5. if (firstwrite)
    6. firstwrite = false;
    7. else
    8. cout<<‘ ‘; // 顶点间用空格分开
    9. cout << root[L][r]; // 输出子树的根
    10. preorder(L, root[L][r]-1); // 前序遍历左子树
    11. preorder(root[L][r]+1, r); // 前序遍历右子树
    12. }

    【参考程序片段】{Pascal版}

    1. procedure preorder(L, r: integer);
    2. // 前序遍历顶点L..顶点r对应的子树
    3. begin
    4. if L > r then exit;
    5. if firstwrite then
    6. firstwrite := false
    7. else
    8. write(‘ ‘); {顶点间用空格分开}
    9. write(way[L, r]); {输出子根}
    10. preorder(L, way[L, r] - 1); {前序遍历左子树}
    11. preorder(way[L, r] + 1, r); {前序遍历右子树}
    12. end;

    【任务三】主程序 有了search函数和preorder过程,就不难得出主程序了:

    『Cpp』

    1. int main()
    2. {
    3. int n, i;
    4. bool firstwrite;
    5. cin >> n; // 读顶点数
    6. for(i=1; i<=n; i++) // 状态转移方程初始化
    7. for(j=i; j<=n; j++)
    8. f[i][j] = -1;
    9. for (i=1; i<=n; i++) {
    10. cin >> f[i][i]; // 读顶点i的分值
    11. root[i][i] = i; // 顶点i单独成一棵子树
    12. }
    13. cout << search(1, n) << endl; // 计算和输出最高加分
    14. firstwrite = true; // 设立首顶点标志
    15. preorder(1, n); // 前序遍历二叉树
    16. return 0;
    17. }

    『Pascal』

    1. read(n); {读顶点数}
    2. for i := 1 to n do {状态转移方程初始化}
    3. for j := i to n do
    4. f[i j] := -1;
    5. for i := 1 to n do
    6. begin
    7. read(temp); {读顶点i的分值}
    8. f[i, i] := temp; way[i, i] := i; {顶点i单独成一棵子树}
    9. end;
    10. writeln(search(1, n)); {计算和输出最高加分}
    11. firstwrite := true; {设立首顶点标志}
    12. preorder(1, n); {前序遍历二叉树}
    13. writeln;

    以上算法采用了自上而下的记忆化方法,即程序流程基本按照原问题的递归定义,不同的是,它专门设置了一张表way,用来记忆在求解过程中得出的所有子树的根,以便在前序遍历中使用。 另外,当在递归过程中第一次遇到一个子问题(f[L, r]=-1)时,计算其解,以后每遇到该子问题时,就不重复计算f[L, r]了。记忆化方式有着只解那些肯定要解的子问题的优点。_在应用动态规划方法求解非线性结构问题时,一般采用自上而下的记忆化方法_。

  • 相关阅读:
    Webpack中的Loader和Plugin:理解与使用
    Java基础,transient关键字
    [JS真好玩] 掘金创作者必备: 监控每天是谁取关了你?
    深析SpringCloud与Docker微服务架构
    c编译器学习07:minilisp编译器改造(debug模式支持调试)
    全国366个市县日度空气质量数据(2016-2020年)(AQI,SO2,NO2,PM2.5,PM10)
    基于5G智能网关的水泵远程监控系统方案
    大数据课程M1——ELK的概述
    是什么可以在Github上标星86.4K,揭秘蚂蚁金服开源的Springboot项目笔记
    【Nginx】基本使用及配置-项目部署
  • 原文地址:https://blog.csdn.net/You_are_hanson/article/details/126083329