1104. 二叉树寻路(python)

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

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

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

给你树上某一个节点的标号 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]