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 存入哈希 map
• dfs 遍历

class Solution {unordered_map<int,int> m;//id,valint 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开始dfsreturn 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

---

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

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