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

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,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

---

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

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