leetcode1104. 二叉树寻路

题目链接：https://leetcode-cn.com/problems/path-in-zigzag-labelled-binary-tree/

题意：

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

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

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

 

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

方法：维护一个数组，存储每一层最左侧和最右侧的和，然后在每一层的时候，判断要不要将数字反转一下，如果需要反转，就从数组中取值，通过求差，找到另外一个数 

 

class Solution {
public:
    vector<int> pathInZigZagTree(int label) {
        //首先确定在第几层
        int depth=0,tmp=label;
        vector<int> ret;//要返回的向量

        while(tmp!=0)
        {
            tmp/=2;
            depth++;
        }

        int floor[21]={0};//总共20层，统计每一层最左侧和最右侧的元素和
        int num=1;
        for(int power=0;power<20;power++)
        {   
            floor[power] = num*2+num-1;
            num*=2;
        }
        ret.emplace_back(label);
        if(depth%2==0)  label = floor[depth-1]-label;
        for(int i=depth-1;i>0;i--)//遍历每一层，找到所需的元素
        {
            label/=2;
            if(i%2==0) 
                ret.emplace_back(floor[i-1]-label);
            else
                ret.emplace_back(label);
        }
        reverse(ret.begin(),ret.end());//反转
        return ret;
    }
};