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

题目描述

小红面对一棵带边权的树,每个节点都有一个非负收益。她可以选择任意节点放置一枚影响半径为 k 的装置,与放置点的树上距离不超过 k 的所有节点都会贡献收益。
请选择放置点,使总收益最大。

输入输出

输入描述
第一行输入节点数 n 和影响半径 k 。
第二行输入 n 个整数 g_0,g_1,...,g_n-1 。
接下来 n-1 行,每行输入 u,v,d ,表示节点 u,v 之间有一条长度为 d 的边。
保证 1 ≤ n ≤ 100 , 1 ≤ k ≤ 10^8 , 0 ≤ g_i,d ≤ 10^5 ,输入图是一棵树。
输出描述
输出能够获得的最大总收益。

样例共 2 组

样例 1 · 选择合适节点后,半径范围内节点收益之和为 640。
输入
7 3
10 10 10 300 10 10 300
0 1 1
0 2 1
1 3 3
1 4 1
2 5 3
2 6 1
输出
640
样例 2 · 最优放置点可覆盖收益总和为 300 的节点集合。
输入
5 2
200 100 250 100 100
1 0 20
0 2 20
3 1 2
4 1 2
输出
300

算法解析依据充分

考点:树

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

题目画像

  • 数据规模:n ≤ 100
  • 元素值域:k ≤ 100000000,g_i ≤ 1e5,d ≤ 1e5(注意整数类型选择,避免溢出)
  • 复杂度门槛:允许 O(n³) ~ O(n⁴) 的多重循环,可以放心枚举。
  • 源站时限:1 秒(牛客口径,非本题专属门槛)

解题思路

推荐方向:树

本题切入点

n≤100 极小,直接枚举放置点:对每个候选点从它出发 DFS 累加树上距离 ≤k 的节点收益取最大(也可用换根 DP 降到 O(n))。

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

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

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

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

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

该范式的通法易错点

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

对照本题

  • 数据规模 n ≤ 100,允许 O(n³) ~ O(n⁴) 的多重循环,可以放心枚举。
  • 元素值域最大到 100000000 —— 求和 / 相乘时记得开 64 位整数。

样例解读

样例 1:输入 7 3 / 10 10 10 300 10 10 300 / 0 1 1 / 0 2 1 / 1 3 3 / 1 4 1 / 2 5 3 / 2 6 1 → 输出 640

选择合适节点后,半径范围内节点收益之和为 640。

样例 2:输入 5 2 / 200 100 250 100 100 / 1 0 20 / 0 2 20 / 3 1 2 / 4 1 2 → 输出 300

最优放置点可覆盖收益总和为 300 的节点集合。

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

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

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