【算法刷题】二叉树的黄金指数之和（DFS深度优先搜索）-北京尧图网络

【算法刷题】二叉树的黄金指数之和DFS深度优先搜索平台蓝桥网 / 算法竞赛考点二叉树遍历、深度优先搜索DFS、数据类型溢出防护、1-based 下标映射 1. 题目描述给定一棵包含nnn个节点的二叉树节点编号为1∼n1 \sim n1∼n其中111号节点为根节点。第iii个节点的权重为wiw_iwi。请你计算出这棵树中黄金指数为000的所有节点的权重之和。黄金指数定义根节点的黄金指数为000。若一个节点是其父节点的左儿子则它的黄金指数父节点的黄金指数1 11。若一个节点是其父节点的右儿子则它的黄金指数父节点的黄金指数−1- 1−1。 2. 解题思路结构存储使用数组left_child[i]和right_child[i]存储每个节点iii的左右子节点编号如果值为000表示对应位置为空。使用数组weight[i]存储节点iii的权重。DFS 状态传递从根节点111开始递归函数定义为dfs(u, gold_index)其中u为当前节点编号gold_index为到达当前节点时的黄金指数。每访问到一个节点若gold_index 0则将当前节点权重weight[u]累加至全局变量ans中。向左递归遍历时指数传递为gold_index 1向右递归遍历时指数传递为gold_index - 1。复杂度和防错策略时间复杂度O(n)\mathcal{O}(n)O(n)每个节点仅访问一次。空间复杂度O(n)\mathcal{O}(n)O(n)主要为递归栈深度与树的存储空间。数据类型节点权重累加和ans需使用long long类型避免多节点权重累加时发生整型溢出。⚠️ 3. 易错点总结数组下标与编号对齐1-based Indexing节点编号为1∼n1 \sim n1∼n输入循环必须从i1i 1i1到ini nin切勿使用i0i 0i0到in−1i n - 1in−1否则会导致权重与节点编号错位。累加对象错误当判定gold_index 0时应该加的是weight[u]而非gold_index。右子树的方向计算往右走是黄金指数−1-1−1不要误写成111。 4. C 完整代码#includeiostreamusingnamespacestd;constintMAXN100005;intweight[MAXN];intleft_child[MAXN];intright_child[MAXN];longlongans0;// 存储权重总和防止爆 int// DFS 深度优先搜索voiddfs(intu,intgold_index){if(u0)return;// 当黄金指数为 0 时累加当前节点的权重if(gold_index0){answeight[u];}// 遍历左子树黄金指数 1if(left_child[u]!0){dfs(left_child[u],gold_index1);}// 遍历右子树黄金指数 -1if(right_child[u]!0){dfs(right_child[u],gold_index-1);}}intmain(){// 开启 IO 优化提升读写效率ios::sync_with_stdio(false);cin.tie(nullptr);intn;if(!(cinn))return0;// 读取每个节点的权重下标从 1 到 nfor(inti1;in;i){cinweight[i];}// 读取左右儿子节点下标从 1 到 nfor(inti1;in;i){cinleft_child[i]right_child[i];}// 从根节点 1 开始遍历初始黄金指数为 0dfs(1,0);// 输出最终答案coutans\n;return0;}

【算法刷题】二叉树的黄金指数之和（DFS深度优先搜索）

相关推荐

xAI 新一代旗舰大模型Grok 4.7深度评测：同价翻倍的编码与知识工作旗舰，长任务能力跃升

STM32驱动MA730/MT6835磁编SPI通信实战避坑指南

Python二维码生成器程序运行时报错 ModuleNotFoundError: No module named ‘PIL‘ 解决方法

更多资讯

Dopamine 实验统计汇总实战：深入解析 colab.utils.summarize_data 逐迭代数据聚合

EMQX MQTT 桥接陈旧连接状态修复解析：从「假 Connected」到真实健康检查与自动重连

CAD软件有哪些？建筑、机械、协同各领域代表软件梳理

AI陪伴机器人DTO参数校验-把脏数据拦在门口

海信电视秋季新品打出王炸，双7旗舰首发，RGB-Mini LED电视体验常用常新

Python coding + ML + general coding ability

今日更新

Play Framework 迁移指南：移除 GlobalSettings，全面转向依赖注入（Scala 与 Java）

Numba 类型推断机制详解：从 Numba IR 到编译期类型重建的完整原理与实践

企业级项目dragonballz_e159-1的技术架构与实现方案

本周精选

深入解析Transformer多头注意力机制与工程优化

OpenClaw 的 Skills 跑学习任务，模型通道改到 TaoToken 通道行不行？

ChatGPT报错Oops, an error occurred! 全链路排查指南

本月热门

基于YOLO的焊缝缺陷检测：从数据标注到工业部署全流程实践

分布式服务的决策记录

创业产品的首版范围

读完文章，想聊聊您的网站？