Leetcode1104。バイナリツリーパスファインディング

Leetcode1104。バイナリツリーパスファインディング

トピック

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

如下图所示，在奇数行（即，第一行、第三行、第五行……）中，按从左到右的顺序进行标记；

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

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

例1：

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

例2：

输入：label = 26
输出：[1,2,6,10,26]

アイデア

• まず、与えられた範囲は10 ^ 6であるため、この質問をツリーの観点から考慮する必要はありません。
• このトピックのツリーは完全なバイナリツリーである必要があります。通常のツリーの場合、親ノードラベル=子ノードラベル// 2
• このトピックのツリーはジグザグ走査であるため、法則を注意深く観察し、親ノードラベル=（子ノードのレイヤーの最大値+レイヤーの最小値-子ノードラベル）// 2
• したがって、最初に現在のラベルがどのレイヤーであるかを見つけることができます
• 次に、レイヤーごとにトラバースします

コード

class Solution:
    def pathInZigZagTree(self, label):
        floor = 1

        while 2 ** floor - 1 < label:
            floor += 1

        res = []

        for i in range(floor, 0, -1):
            res.append(label)
            label = (2 ** i - 1 + 2 ** (i - 1) - label) // 2

        return res[-1::-1]