1104. 二叉树寻路

1104. 二叉树寻路

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

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

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

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

示例 1：

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

示例 2：

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

思路

再利用「每层节点数量翻倍」&「隔层奇偶性翻转」，寻址出上一层的节点下标（令每层下标均「从左往右」计算，并从 1开始），直到构造出答案

class Solution {
   // 第 level 层的起始节点值
   int getStart(int level) {
       return (int)Math.pow(2, level - 1);
   }
   // 第 level 层的结束节点值
   int getEnd(int level) {
       int a = getStart(level);
       return a + a - 1;
   }
   public List<Integer> pathInZigZagTree(int n) {
       // 计算 n 所在层级
       int level = 1;
       while (getEnd(level) < n) level++;

       int[] ans = new int[level];
       int idx = level - 1, cur = n;
       while (idx >= 0) {
           ans[idx--] = cur;
           int tot = (int)Math.pow(2, level - 1);
           int start = getStart(level), end = getEnd(level);
           if (level % 2 == 0) {
               // 当前层为偶数层，则当前层节点「从右往左」数值递增，相应计算上一层下标也应该「从右往左」
               int j = tot / 2;
               for (int i = start; i <= end; i += 2, j--) {
                   if (i == cur || (i + 1) == cur) break;
               }
               int prevStart = getStart(level - 1);
               while (j-- > 1) prevStart++;
               cur = prevStart;
           } else {
               // 当前层为奇数层，则当前层节点「从左往右」数值递增，相应计算上一层下标也应该「从左往右」
               int j = 1;
               for (int i = start; i <= end; i += 2, j++) {
                   if (i == cur || (i + 1) == cur) break;
               }
               int prevEnd = getEnd(level - 1);
               while (j-- > 1) prevEnd--;
               cur = prevEnd;
           }
           level--;
       }
       List<Integer> list = new ArrayList<>();
       for (int i : ans) list.add(i);
       return list;
   }
}