LeetCode 1104. 二叉树寻路

1104. 二叉树寻路

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

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

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

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

示例 1：

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

示例 2：

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

提示：

1 <= label <= 10^6

题解：先初始化，根据题目的要求把每个节点上对应的数字进行赋值，在赋值过程中遇到我们需要搜索的目标数字就标记该数字在二叉树中的位置pos，找到后就直接不断pos/=2一直让pos=1返回根节点。

AC代码

class Solution {
public:
int a[10001000];
vector<int> pathInZigZagTree(int label) {
	int ans=1;
	int pos;
	for(int i=1;i<=30;i++)
	{
		int l=pow(2,i-1),r=pow(2,i)-1;
		if(i%2==1)
		{
			for(int j=l;j<=r;j++)
			{
				a[j]=ans;
				if(ans==label)
				pos=j;
				ans++;
			}			
		}
        else
        {
            for(int j=r;j>=l;j--)
            {
                a[j]=ans;
                if(ans==label)pos=j;
                ans++;
            }
        }
        if(l>label)break;
	}
	vector<int>q,res;
	while(true)
	{
		q.push_back(a[pos]);
		if(pos==1)break;
		pos/=2;
	}
    for(int i=q.size()-1;i>=0;i--)
    res.push_back(q[i]);
	return res;
}
};