华为 · 树 · 算法编程题
华为 K ≤ 1e3 时限 1 秒 / 256 MB

题目描述

小红用一棵二叉依赖树管理资源。节点给出自身增量,一个节点的完整大小等于从根到该节点路径上所有增量之和。若某节点完整大小不大于 0,该节点及其整棵子树都被剪枝。
给出互不相同的节点增量所构成的前序遍历和中序遍历,请恢复树、完成剪枝,并将剩余节点中最大的 K 个完整大小按升序输出。

输入输出

输入描述
第一行输入前序遍历,第二行输入中序遍历,第三行输入 K 。两行长度相同。
保证节点数在 [1,1000] ,节点值互不相同且在 [-10000,10000] , 1≤ K≤1000 。
输出描述
按升序输出最大的 K 个完整大小;若剪枝后为空,输出 `null`。若剩余节点少于 K 个,则输出全部。

样例共 2 组

样例 1 · 各节点完整大小为 8、10、5、14、10,取最大的三个。
输入
8 2 -3 9 5
2 8 9 -3 5
3
输出
10 10 14
样例 2 · 完整大小不大于 0 的节点连同子树被剪枝。
输入
1 2 3 -5 5 6
2 1 3 5 -5 6
1
输出
4

算法解析依据一般

考点:树

数据规模 K ≤ 1e3 | 限制 1 秒 / 256MB | 标准输入输出

题目画像

  • 数据规模:K ≤ 1e3
  • 复杂度门槛:允许 O(n²);若 n ≤ 300,O(n³) 通常也跑得动,需看具体常数。
  • 源站时限:1 秒(牛客口径,非本题专属门槛)

解题思路

参考方向:树

树结构上的遍历(DFS 求子树信息 / BFS 求层序),多数树题是「后序遍历 + 回溯」。

思路框架(树 通法 · 非本题专属)

  1. 按输入建图(父子关系一般给父节点编号,孩子用邻接表存)。
  2. 确定遍历方式:自底向上合并子树信息(后序 DFS)或自顶向下传参(前序 DFS)。
  3. 对每个节点,用孩子的答案合并出当前节点的答案。
  4. 涉及层间操作(如层序变换)时用 BFS 逐层处理。

实现要点:递归深度可能到 1e5,注意递归爆栈;必要时改迭代或调整递归深度。

复杂度:时间 O(n) | 空间 O(n)

该范式的通法易错点

  • 递归深度过大导致爆栈。
  • 建树时父子关系方向搞反(把树当成有向图但遍历方向错)。

对照本题

  • 数据规模 K ≤ 1e3,允许 O(n²);若 n ≤ 300,O(n³) 通常也跑得动,需看具体常数。
本题在源数据中没有官方考点标签,方向由题面特征推断,仅供参考。

样例解读

样例 1:输入 8 2 -3 9 5 / 2 8 9 -3 5 / 3 → 输出 10 10 14

各节点完整大小为 8、10、5、14、10,取最大的三个。

样例 2:输入 1 2 3 -5 5 6 / 2 1 3 5 -5 6 / 1 → 输出 4

完整大小不大于 0 的节点连同子树被剪枝。

解析由校招宝本地引擎整理(依据源站考点标签 / 人工判题标注 / 题面规模信号),非官方题解,仅供思路参考。

本题来源:2026年-华为-06月24号开发岗。

‹ 上一题 全部编程题 下一题 ›
编程算法题为只读内容:无需作答,直接看题与解析 · 本站不提供在线判题 · 解析由校招宝本地引擎整理,非官方题解