leetcode1104. 二叉树寻路

题目描述

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

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

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

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

示例 1：

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

示例 2：

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

题解

每次找出label距离二叉树最右侧的位置，然后算出两者之差diff。每上升一层diff除以2，并可以利用diff和当前行的奇偶性求出路径上的每一个值。

class Solution {
    public List<Integer> pathInZigZagTree(int label) {
        //求label在第几行
        int num = (int)(Math.log(label) / Math.log(2)) + 1;
        LinkedList<Integer> result = new LinkedList<>();
        result.addFirst(label);
        //diff为距离右侧的个数
        int diff;
        if(num % 2 == 0) {
            diff = label - (int)Math.pow(2, num - 1);
        }
        else {
            diff = (int)Math.pow(2, num) - 1 - label;
        }
        num--;
        while(num > 0) {
            if(num % 2 == 0) {
                label = diff / 2 + (int)Math.pow(2, num - 1);
            }
            else {
                label = (int)Math.pow(2, num) - 1 - diff / 2;
            }
            diff = diff / 2;
            result.addFirst(label);
            
            num--;
        }
        return result;
    }

}