LeetCode Hard:子树最小缺失基因值算法解析

📅 2026/7/18 12:28:43
LeetCode Hard:子树最小缺失基因值算法解析
1. 问题背景与核心概念解析这道LeetCode Hard题目描述了一棵包含n个节点的树每个节点被赋予一个独一无二的基因值取值范围1到n。我们需要为每棵子树计算其内部缺失的最小基因值。子树定义为以某节点为根的所有后代节点集合。理解这个问题的关键在于把握三个核心特性基因值唯一性所有节点的基因值互不相同且正好覆盖1到n的整数范围子树定义以任意节点为根的子树包含该节点及其所有后代节点最小缺失值对于子树内基因值集合找出不在其中的最小正整数举个例子假设某子树包含基因值{2,3,5}那么缺失值就是1若包含{1,2,3}则缺失值为4。这个定义与经典的缺失的第一个正整数问题类似但在树结构上需要高效处理。2. 关键结论与算法思路2.1 三大核心结论通过分析题目特性可以得出三个决定性结论无1子树定理如果子树中不包含基因值1那么该子树的答案必定是1。因为1是最小的正整数且题目保证基因值在1到n之间。有1路径定理由于基因值唯一包含1的节点会形成一条从根节点到某个叶节点的路径称为1-path。只有这些节点所在的子树可能需要计算大于1的答案。连续区间定理对于在1-path上的节点其子树的答案必然是当前已见最大基因值1。因为基因值唯一且连续缺失值会形成连续的缺口。2.2 算法框架设计基于这三个结论我们可以设计出高效算法预处理阶段通过一次DFS遍历记录每个节点是否包含基因值1并构建父指针关系用于回溯。主处理阶段对于不包含1的子树直接返回1对于包含1的子树沿着1-path收集基因值动态维护当前缺失的最小值优化技巧使用哈希集合来记录已访问的基因值将时间复杂度优化到O(n)。3. 详细实现与代码解析3.1 数据结构准备def smallestMissingValueSubtree(parents, nums): n len(parents) # 构建树结构 tree [[] for _ in range(n)] for i in range(1, n): tree[parents[i]].append(i) # 初始化结果数组 res [1] * n # 找到基因值为1的节点 one_pos -1 for i in range(n): if nums[i] 1: one_pos i break # 如果没有1所有子树答案都是1 if one_pos -1: return res3.2 DFS标记包含1的子树# 标记包含1的子树 has_one [False] * n def dfs_mark(u): has_one[u] nums[u] 1 for v in tree[u]: dfs_mark(v) has_one[u] has_one[u] or has_one[v] dfs_mark(0)3.3 处理1-path上的节点# 处理包含1的子树 visited set() current_missing 1 u one_pos while u ! -1: # 收集当前节点的基因值 stack [u] while stack: node stack.pop() if nums[node] not in visited: visited.add(nums[node]) # 更新current_missing while current_missing in visited: current_missing 1 # 添加子节点只添加不包含1的子树 for v in tree[node]: if not has_one[v]: stack.append(v) res[u] current_missing u parents[u] # 回溯到父节点4. 复杂度分析与优化证明4.1 时间复杂度DFS标记阶段每个节点被访问一次O(n)主处理阶段每个基因值最多被加入集合一次每个缺失值检查最多执行n次1到n递增总体O(n)总时间复杂度O(n)4.2 空间复杂度树结构存储O(n)哈希集合最坏情况存储所有基因值O(n)递归栈深度最坏情况树退化为链表O(n)总空间复杂度O(n)关键优化点只有在1-path上的节点需要处理基因值收集其他节点可以直接返回1。这使得算法在实践中非常高效。5. 常见错误与调试技巧5.1 典型错误模式错误假设基因值范围忽视题目中基因值正好是1到n的排列导致不必要的排序操作。重复计算缺失值在遍历过程中没有有效维护当前缺失值每次都从头计算导致O(n^2)复杂度。子树包含判断错误错误判断哪些子树包含基因值1导致漏处理或过度处理。5.2 调试建议小规模测试用例构造包含3-5个节点的树手工计算预期结果验证代码。边界情况检查根节点基因值为1所有节点基因值都大于1基因值完全升序/降序排列可视化工具使用树可视化工具如Graphviz绘制树结构辅助理解处理流程。6. 算法扩展与变种思考6.1 基因值不唯一的情况如果题目放宽基因值唯一性的限制我们需要使用频率统计代替集合存储缺失值计算改为寻找最小未出现正整数时间复杂度可能提高到O(nlogn)需要排序6.2 多查询场景优化如果需要处理多个不同的基因值查询而不仅仅是1可以考虑预处理所有节点的基因值集合使用合并-查找数据结构维护子树关系采用离线算法批量处理查询6.3 大规模数据应用当树规模极大时如1e6节点改用迭代DFS避免栈溢出使用位图代替哈希集合节省空间考虑并行处理不同子树这个问题的解法展示了如何利用问题本身的特殊性质基因值唯一、范围限定来设计高效算法。在实际工程中这种观察特性→得出结论→设计算法的思维方式比单纯记忆算法模板更为重要。