1104. 二叉树寻路(python)

标签: python  leetcode

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

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

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

给你树上某一个节点的标号 label,请你返回从根节点到该标号为 label 节点的路径,该路径是由途经的节点标号所组成的。
在这里插入图片描述

示例 1:

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

示例 2:

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

提示:

1 <= label <= 10^6

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

思路: 从低到上,依次将label除以2即可得到对应父节点,但由于偶数行和奇数行顺序相反,如果label初始时在奇数行,则求得的父节点要在偶数行置反,反之要置反其奇数行。置反时,由于他们的对称关系可知,左边某节点其左边界的距离等于其置反节点在右边界的距离。可得求置反结点的公式t = pow(2, i ) -1 - tmp[i - 1] + pow(2, i - 1),t为tmp[i - 1]的置反节点,,i为层数, tmp[i - 1]为要置反的节点,pow(2, i ) -1为右边界,pow(2, i - 1)为左边界 。因为深度为k的完全二叉树的结点个数为pow(2, i) - 1个, 所以可求出置反节点所在行的节点取值范围为pow(2, i) - 1 - pow(2, i - 1) - 1。

代码如下:

class Solution(object):
    def pathInZigZagTree(self, label):
        """
        :type label: int
        :rtype: List[int]
        """
        if label == 1:
            return [1]
        tmp = []
        while label > 0:
            tmp.append(label)
            label //= 2
        tmp = tmp[::-1]
        #print(tmp)
        n = len(tmp)
        flg = n % 2 # 初始label偶数行,置反奇数行位置,反之置反偶数行位置
        for i in range(1, n):
            if i % 2 == flg:
                t = pow(2, i + 1) -1 - tmp[i] + pow(2, i)
                tmp[i] = t
        print(tmp)
        return tmp
        


x = Solution()
x.pathInZigZagTree(26)

输出结果: [1, 2, 6, 10, 26]

原文链接:加载失败,请重新获取