Leetcode 1104.二叉树寻路
1 题目描述(Leetcode题目链接)
在一棵无限的二叉树上,每个节点都有两个子节点,树中的节点 逐行 依次按 “之” 字形进行标记。
如下图所示,在奇数行(即,第一行、第三行、第五行……)中,按从左到右的顺序进行标记;而偶数行(即,第二行、第四行、第六行……)中,按从右到左的顺序进行标记。
给你树上某一个节点的标号 label,请你返回从根节点到该标号为 label 节点的路径,该路径是由途经的节点标号所组成的。
2 题解
首先假设这课无限的二叉树是按照一行一行顺序排列的,通过完全二叉树父子节点之间的关系,我们可以很容易地得到一个节点 i i i的父节点是 i 2 \cfrac{i}{2} 2i。比如 14 14 14和 15 15 15的父节点应该是 7 7 7,那么现在二叉树是蛇形排列,那就说明,节点父节点是关于二叉树对称的另外一个节点,所以这里 14 14 14和 15 15 15的父节点是 4 4 4。这样我们就可以通过不断地找对称来做了。
class Solution:
def pathInZigZagTree(self, label: int) -> List[int]:
n = int(math.log2(label))
retv = [label]
while n >= 1:
t = retv[-1]//2
retv.append(2**n + 2**(n-1) - t - 1)
n -= 1
return retv[::-1]