LeetCode 1104. 二叉树寻路

最新推荐文章于 2022-02-22 00:04:05 发布

_almost__

最新推荐文章于 2022-02-22 00:04:05 发布

阅读量61

点赞数

分类专栏：刷题

本文链接：https://blog.csdn.net/almost_afei/article/details/119208719

版权

刷题专栏收录该内容

293 篇文章 0 订阅

订阅专栏

难度：中等。
标签：二叉树，数学。

从label向上找parent。

正确解法：

class Solution {

    int getParent(int label, int row){
        int row_begin = pow(2, row);
        int row_end = row_begin * 2 - 1;
        int now_index = row_end - label;
        if(row % 2 == 0){    // ou
            now_index = label - row_begin;
        }
        int parent_index = now_index / 2;
        int parent = row_begin / 2 + parent_index;
        if(row % 2 == 0){   // parent ji
            parent = row_begin - 1 - parent_index;
        }
        return parent;
    }

public:
    vector<int> pathInZigZagTree(int label) {
        int row = log(label) / log(2);
        vector<int> ans(row + 1);
        ans[row] = label;
        while(label != 1){
            label = getParent(label, row);
            row--;
            ans[row] = label;
        }
        return ans;
    }
};