81 lines
3.1 KiB
Markdown
81 lines
3.1 KiB
Markdown
## 题目地址
|
||
|
||
https://leetcode-cn.com/problems/path-in-zigzag-labelled-binary-tree/description/
|
||
|
||
## 题目描述
|
||
|
||
在一棵无限的二叉树上,每个节点都有两个子节点,树中的节点 逐行 依次按 “之” 字形进行标记。
|
||
|
||
如下图所示,在奇数行(即,第一行、第三行、第五行……)中,按从左到右的顺序进行标记;
|
||
|
||
而偶数行(即,第二行、第四行、第六行……)中,按从右到左的顺序进行标记。
|
||
|
||
![](https://tva1.sinaimg.cn/large/006tNbRwgy1gaihhb1ysbj310p0gu3zx.jpg)
|
||
|
||
给你树上某一个节点的标号 label,请你返回从根节点到该标号为 label 节点的路径,该路径是由途经的节点标号所组成的。
|
||
|
||
示例 1:
|
||
|
||
输入:label = 14
|
||
输出:[1,3,4,14]
|
||
示例 2:
|
||
|
||
输入:label = 26
|
||
输出:[1,2,6,10,26]
|
||
|
||
提示:
|
||
|
||
1 <= label <= 10^6
|
||
|
||
## 思路
|
||
|
||
假如这道题不是之字形,那么就会非常简单。 我们可以根据子节点的 label 轻松地求出父节点的 label,公示是 label // 2(其中 label 为子节点的 label)。
|
||
|
||
如果是这样的话,这道题应该是 easy 难度,代码也不难写出。我们继续考虑之字形。我们不妨先观察一下,找下规律。
|
||
|
||
![](https://tva1.sinaimg.cn/large/006tNbRwly1gaihn0ktanj30lu093gme.jpg)
|
||
|
||
以上图最后一行为例,对于 15 节点,之字变换之前对应的应该是 8 节点。14 节点对应的是 9 节点。。。
|
||
|
||
全部列举出来是这样的:
|
||
|
||
![](https://tva1.sinaimg.cn/large/006tNbRwly1gaihota82cj30mk0b6wfp.jpg)
|
||
|
||
我们发现之字变换前后的 label 相加是一个定值。
|
||
|
||
![](https://tva1.sinaimg.cn/large/006tNbRwly1gaihpnlpicj309b08dmxl.jpg)
|
||
|
||
因此我们只需要求解出每一层的这个定值,然后减去当前值就好了。(注意我们不需要区分偶数行和奇数行)
|
||
问题的关键转化为求解这个定值,这个定值其实很好求,因为每一层的最大值和最小值我们很容易求,而最大值和最小值的和正是我们要求的这个数字。
|
||
|
||
最大值和最小值这么求呢?由满二叉树的性质,我们知道每一层的最小值就是`2 ** (level - 1)`,而最大值是`2 ** level - 1`。 因此我们只要知道 level 即可,level 非常容易求出,具体可以看下面代码。
|
||
|
||
## 关键点
|
||
|
||
- 满二叉树的性质:
|
||
|
||
1. 最小值是`2 ** (level - 1)`,最大值是`2 ** level - 1`,其中 level 是树的深度。
|
||
2. 假如父节点的索引为 i,那么左子节点就是 2\*i, 右边子节点就是 2\*i + 1。
|
||
3. 假如子节点的索引是 i,那么父节点的索引就是 i // 2。
|
||
|
||
- 先思考一般情况(不是之字形), 然后通过观察找出规律
|
||
|
||
## 代码
|
||
|
||
```python
|
||
class Solution:
|
||
def pathInZigZagTree(self, label: int) -> List[int]:
|
||
level = 0
|
||
res = []
|
||
# for each level, ranged from 2 ** (level - 1) to 2 ** level - 1
|
||
while 2 ** level - 1 < label:
|
||
level += 1
|
||
|
||
while level > 0:
|
||
res.insert(0, label)
|
||
label = 2 ** (level - 1) + 2 ** level - 1 - label
|
||
label //= 2
|
||
level -= 1
|
||
return res
|
||
```
|