加分二叉树(信息学奥赛一本通- P1580)(洛谷-P1040)

一、 题目分析

在信息学奥赛的早期真题中,NOIP 2003 的《加分二叉树》是一道具有代表性的好题。题面看似是在考查二叉树的构建与遍历,但它却给出了一个很致命、也是破局核心的条件: "二叉树的中序遍历为 (1, 2, 3, ..., n)"

在数据结构中,中序遍历的顺序是"左子树 → 根 → 右子树"。既然整棵树的中序遍历是连续的 1 到 n,这就意味着一个物理定律:对于树上的任意一棵子树,它所包含的所有节点编号,在物理上绝对构成一段连续的区间 [i,j]!

一旦清楚这一点,这道题的图论外衣就被彻底扒下,露出了它区间DP的真实面目。

二、 核心状态定义与转移方程

既然是处理连续区间,我们直接套用区间DP的经典模型:

  • 状态定义:设 dp[i][j] 表示由编号i到j的节点所组成的子树,能获得的最高加分。

  • 记号本(状态溯源) :题目不仅要求最高分,还要求输出前序遍历。我们额外开一个数组 root[i][j],记录区间 [i,j] 取得最高分时,是哪个节点k当了树根。

转移策略(枚举断点打擂台): 对于区间 [i,j],我们不知道谁当根节点收益最大。因此,我们让区间内的每一个节点 k (i≤k≤j) 都轮流"坐庄"当一次根节点。 当k为根时,区间被完美切割:左子树是 [i,k−1],右子树是 [k+1,j]。

根据题意"加分=左子树加分×右子树加分+根节点分数",得出状态转移方程:

dp[i][j]=max(dp[i][j],dp[i][k−1]×dp[k+1][j]+a[k])

三、 四个易错点

区间DP的代码骨架很短,但极其容易在初始化和边界上死循环。以下四个坑点,都是校队同学真实出错的地方:

  1. 空子树的合法性(越界陷阱) 当选定最左侧节点i当根时,左子树区间变为 [i,i−1]。这是一个空树。题目规定空子树加分为 1。所以必须初始化 dp[i][i−1]=1。注意,当选定最右侧节点 n 当根时,右子树变为 [n+1,n],所以初始化的循环必须开到n+1以防越界。

  2. 叶子节点的独立性(避免公式误伤) 题目规定"叶子的加分就是叶节点本身的分数"。如果我们让长度len=1的区间也进入转移方程,就会多乘上两个空子树的1,导致分数计算错误。最稳妥的做法是:手动初始化长度为1的区间(dp[i][i]=a[i]且root[i][i]=i),主循环从len=2 开始。

  3. 右端点的当根权 枚举根节点k时,必须是 for(int k=i;k<=j; k++),绝不能漏掉等号。二叉树允许只有左子树没有右子树的偏瘫形态,最右边的节点 j 完全有资格当树根。

  4. 整数溢出(数据类型陷阱) 题目明确说明最高加分可能不超过 4×10^9。这个数值已经超过了32位有符号整型int的极限(约 21.4 亿)。因此,dp数组必须果断开long long,否则虽然信息学奥赛一本通能过,但是实际上如果样例大一点是会出错的。

四、 完整代码

cpp 复制代码
#include <iostream>
using namespace std;
int n;
int a[35];//记录每个节点的原本分数
long long dp[35][35];//dp[i][j]代表i-j区间内最高加分
int root[35][35];//记录i和j区间取得最高分时的最优根节点

//输出前序遍历
void print(int l,int r){
    if(l>r) return;//遇到空节点就返回
    int k=root[l][r];//否则就输出根
    cout<<k<<" ";
    print(l,k-1);//递归左子树
    print(k+1,r);//递归右子树
}

int main(){
    cin>>n;
    for(int i=1;i<=n;i++) cin>>a[i];
    for(int i=1;i<=n+1;i++){//注意循环到n+1防止右侧空子树越界
        dp[i][i]=a[i];//叶子结点的最高加分就是自己本身的分数
        dp[i][i-1]=1;//初始化空节点加分位1
        root[i][i]=i;//初始化每个节点是自己的根,叶子节点当根的只能是它自己,防止递归死循环
    }
    for(int len=2;len<=n;len++){//区间长度从2开始遍历
        //枚举左端点
        for(int i=1;i<=n-len+1;i++){
            //右端点
            int j=i+len-1;
            for(int k=i;k<=j;k++){//根节点
            //状态转移,如果l×r+a大于之前加分,就更新加分
            //然后记录本次ij的最优根节点
                if(dp[i][k-1]*dp[k+1][j]+a[k]>dp[i][j]){
                    dp[i][j]=dp[i][k-1]*dp[k+1][j]+a[k];
                    root[i][j]=k;
                }
            }
        }
    }
    //输出最高加分
    cout<<dp[1][n]<<endl;
    //递归输出先序遍历顺序
    print(1,n);
}
相关推荐
再难也得平2 小时前
力扣54. 螺旋矩阵(Java解法)
算法·leetcode·矩阵
飞Link3 小时前
别再被异常数据骗了:深度解析 TSTD 异常检测中的重构模型(AutoEncoder 实战)
人工智能·算法·重构·回归
小曹要微笑3 小时前
C#中的各种数据类型
算法·c#·数据类型·c#数据类型
靠沿3 小时前
【优选算法】专题九——链表
数据结构·算法·链表
weixin_649555673 小时前
C语言程序设计第四版(何钦铭、颜晖)第七章利用数组判断上三角矩阵
算法
星爷AG I3 小时前
14-4 运动控制理论:协同理论(AGI基础理论)
算法·机器学习·agi
I_LPL3 小时前
day48 代码随想录算法训练营 图论专题1
java·算法·深度优先·图论·广度优先·求职面试
absunique4 小时前
多路归并算法在外部排序中的实现与优化的技术7
算法
鹿鸣悠悠4 小时前
【AI-08】Prompt(提示词)
人工智能·算法