leetcode- 1104. 二叉树寻路

在一棵无限的二叉树上,每个节点都有两个子节点,树中的节点 逐行 依次按 “之” 字形进行标记。

如下图所示,在奇数行(即,第一行、第三行、第五行……)中,按从左到右的顺序进行标记;

而偶数行(即,第二行、第四行、第六行……)中,按从右到左的顺序进行标记。



给你树上某一个节点的标号 label,请你返回从根节点到该标号为 label 节点的路径,该路径是由途经的节点标号所组成的。

 

示例 1:

输入:label = 14
输出:[1,3,4,14]

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/path-in-zigzag-labelled-binary-tree
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

每层节点数目和 节点起始值与结束值不变
对于给定的值我们只需要计算他所在的层数 以及在对应层位于从左到右第几个节点 确定好后再根据奇数层/偶数层计算对应值即可

class Solution(object):

    def pathInZigZagTree(self, label):
        """
        :type label: int
        :rtype: List[int]
        """
        level = 0
        while label > ((1 << level) - 1):
            level += 1
        res = []
        if level % 2 == 1:
            start_index = label - (1 << (level - 1))
        else:
            start_index = (1 << level) - label - 1
        self.depth(level - 1,  start_index , res)
        return res
	
	# level 从 0 开始
	# index 从左到右第n个节点 0,1,2,3...
	# res 结果列表
    def depth(self, level, index, res):
        if level == 0:
            res.insert(0, 1)
            return
        if level % 2 == 0:
            curr_val = (1 << level) + index
        else:
            curr_val = (1 << level) + ((1<< level) - index - 1)
        res.insert(0, curr_val)
        self.depth(level - 1, int(index / 2), res)

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值