在一棵无限的二叉树上,每个节点都有两个子节点,树中的节点 逐行 依次按 “之” 字形进行标记。
如下图所示,在奇数行(即,第一行、第三行、第五行……)中,按从左到右的顺序进行标记;
而偶数行(即,第二行、第四行、第六行……)中,按从右到左的顺序进行标记。
给你树上某一个节点的标号 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)