JAVA程序设计：二叉树寻路（LeetCode：1104）

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

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

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

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

 

示例 1：

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

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

提示：

1 <= label <= 10^6

思路：可以发现，若是一棵正常树，则路径一定是label不停除以2得到的数，例如找到14的路径是：1，3,7,14。而对于本题之字形的树很容易找到规律，需要翻转的节点所在层一定到label最初那一层的距离为奇数。

class Solution {
    public List<Integer> pathInZigZagTree(int label) {

        int now = 0, sum = 0;
        List<int[]> list = new ArrayList<>();
        List<Integer> res = new ArrayList<>();
        List<Integer> ans = new ArrayList<>();

        while (now < 30) {
            list.add(new int[]{sum + 1, sum + (1 << now)});
            sum += (1 << now);
            now++;
        }

        int depth = 0;
        for (int i = 0; i < 30; i++) {
            if (label >= list.get(i)[0] && label <= list.get(i)[1]) {
                depth = i;
                break;
            }
        }

        while (label > 0) {
            res.add(label);
            label /= 2;
        }

        Collections.reverse(res);

        int size = res.size();
        for (int i = 0; i < size - 1; i++) {
            if (i % 2 != (size - 1) % 2)
                ans.add(list.get(i)[0] + list.get(i)[1] - res.get(i));
            else
                ans.add(res.get(i));
        }

        ans.add(res.get(size - 1));

        return ans;

    }
}