LeetCode

LeetCode 1104. 二叉树寻路

2020-03-05  本文已影响0人  桐桑入梦

在一棵无限的二叉树上,每个节点都有两个子节点,树中的节点 逐行 依次按 “之” 字形进行标记。如下图所示,在奇数行(即,第一行、第三行、第五行……)中,按从左到右的顺序进行标记;而偶数行(即,第二行、第四行、第六行……)中,按从右到左的顺序进行标记。

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

示例 1:

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

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

提示:

2**1  - 1 = 1
2**2  - 1 = 3 
2**3  - 1 = 7
2**4  - 1 = 15
...
2**10 - 1 = 1023
...
2**20 - 1 = 1024 * 1024 -1 > 1000000
class Solution {

    public static final int[] nums = new int[21];

    static{
        //计算i层完美二叉树的节点总数
        nums[0] = 1;
        for( int i = 1; i <= 20; i++ )
            nums[i] = nums[i - 1] * 2;
        for( int i = 1; i <= 20; i++)
            nums[i] -= 1;
    }

    public List<Integer> pathInZigZagTree(int label) {
        //计算高度
        int level = 1;
        while( label > nums[level] ) level++;

        //判断奇偶,添加路径
        List<Integer> res = new ArrayList<Integer>();
        while( level > 0 ){
            res.add( label );
            //这里转换一下标号,分奇偶,发现一样的思路
            label = ( nums[ level - 1 ] + ( nums[level] + 1 - label ) ) / 2;
            level--; 
        }
        Collections.reverse(res);
        return res;
    }
}
效率挺高,就是使用了一个数组内存有消耗
上一篇下一篇

猜你喜欢

热点阅读