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

Michael阿明

发布于 2020-07-13 16:32:53

1K0

发布于 2020-07-13 16:32:53

文章被收录于专栏：Michael阿明学习之路

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. 解题

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

本文参与腾讯云自媒体同步曝光计划，分享自作者个人站点/博客。

原始发表：2020/07/04 ，如有侵权请联系 cloudcommunity@tencent.com 删除

二叉树