前言
很久没写博客了,最近在刷题,写了个简单的双百题目,于是来分享记录一下。
题目
在一棵无限的二叉树上,每个节点都有两个子节点,树中的节点 逐行 依次按 “之” 字形进行标记。
如下图所示,在奇数行(即,第一行、第三行、第五行……)中,按从左到右的顺序进行标记;
而偶数行(即,第二行、第四行、第六行……)中,按从右到左的顺序进行标记。
给你树上某一个节点的标号 label,请你返回从根节点到该标号为 label 节点的路径,该路径是由途经的节点标号所组成的。
示例 1:
输入:label = 14
输出:[1,3,4,14]
示例 2:
输入:label = 26
输出:[1,2,6,10,26]
未逆序处理的二叉树
正常未逆序处理的二叉树寻路,从下至上,下一个节点都是上一个节点的二分之一
逆序处理的二叉树
经过逆序处理的二叉树,每个上下两层的关系相当于是进行了逆序处理,
因此对于上一层的逆序反转,我们需要找到对应反转的数字。
容易观察到的是,找到反转前后的两个数字,a、b, a+b = (当前层最小值+当前层最大值)。
因此我们在知道a的情况下,计算b只需要知道当前层的最大与最小值,这两个值仅依赖于二叉树层数。
如lable为14,未逆序处理的二叉树,上一个节点为7,经过逆序处理的节点 = 4(最小)+7(最大)-7(未逆序处理)=4.
题解思路
- 计算lable所在的层级。
- 依次从下至上遍历层级。
- 计算未经逆序处理的上一层数据。
- 根据上一层节点最大、最小值计算出逆序处理的上一层节点数据
public List<Integer> pathInZigZagTree(int label) {
LinkedList<Integer> result = new LinkedList<>();
int level = 0;
int index = 1;
//1.计算label节点对应的二叉树层级
while (label >= index) {
index = index << 1;
level++;
}
//2.依次从下至上添加对应层级节点
while (level > 0) {
result.addFirst(label);
//3.计算未逆序处理的上一层节点数据
label = (label / 2);
//4.已逆序处理的上一层节点=(上一层节点最小值+上一层节点最大值)-未逆序处理的上一层节点数据
label = ((int) Math.pow(2, level - 2) + (int) Math.pow(2, level - 1) - 1) - label;
level--;
}
return result;
}
时间复杂度:log n
空间复杂度:1