LeetCode 666. 路径和 IV（树的遍历）

文章目录

1. 题目2. 解题

1. 题目

对于一棵深度小于 5 的树，可以用一组三位十进制整数来表示。

对于每个整数：

百位上的数字表示这个节点的深度 D，1 <= D <= 4。十位上的数字表示这个节点在当前层所在的位置 P， 1 <= P <= 8。位置编号与一棵满二叉树的位置编号相同。个位上的数字表示这个节点的权值 V，0 <= V <= 9。

给定一个包含三位整数的升序数组，表示一棵深度小于 5 的二叉树， 请你返回从根到所有叶子结点的路径之和。

样例 1: 输入: [113, 215, 221] 输出: 12 解释: 这棵树形状如下: 3 / \ 5 1 路径和 = (3 + 5) + (3 + 1) = 12. 样例 2: 输入: [113, 221] 输出: 4 解释: 这棵树形状如下: 3 \ 1 路径和 = (3 + 1) = 4.

来源：力扣（LeetCode） 链接：https://leetcode-cn.com/problems/path-sum-iv 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

2. 解题

$id=2^{d-1}+p-1$，将 id 和 val 存入哈希 mapdfs 遍历 class Solution { unordered_map<int,int> m;//id,val int sum = 0; public: int pathSum(vector<int>& nums) { if(nums.empty()) return 0; for(int i = 0; i < nums.size(); ++i) { int pos = (1<<(nums[i]/100 -1)) + nums[i]/10%10 -1; m[pos] = nums[i]%10; } dfs(1, 0);//从root编号1，s 为和0开始dfs return sum; } void dfs(int p, int s) { if(!m.count(p)) return; s += m[p]; bool l = m.count(2*p), r = m.count(2*p+1); if(!l && !r) { sum += s; return; } dfs(2*p, s); dfs(2*p+1, s); } };

4 ms 10.2 MB

---

我的博客地址 https://michael.blog.csdn.net/

长按或扫码关注我的公众号（Michael阿明），一起加油、一起学习进步！