前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >LeetCode 666. 路径和 IV(树的遍历)

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 的二叉树, 请你返回从根到所有叶子结点的路径之和。

代码语言:javascript
复制
样例 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. 解题

代码语言:javascript
复制
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 删除

本文分享自 作者个人站点/博客 前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体同步曝光计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
目录
  • 2. 解题
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档