1104. Path In Zigzag Labelled Binary Tree**

1104. Path In Zigzag Labelled Binary Tree**

https://leetcode.com/problems/path-in-zigzag-labelled-binary-tree/

Title Description

In an infinite binary tree where every node has two children, the nodes are labelled in row order.

In the odd numbered rows (ie., the first, third, fifth,…), the labelling is left to right, while in the even numbered rows (second, fourth, sixth,…), the labelling is right to left.

Here Insert Picture Description

Given the label of a node in this tree, return the labels in the path from the root of the tree to the node with that label.

Example 1:

Input: label = 14
Output: [1,3,4,14]

Example 2:

Input: label = 26
Output: [1,2,6,10,26]

Constraints:

  • 1 <= label <= 10^6

C ++ implementation 1

This question needs to be carefully resolved to find the law.

Here Insert Picture Description

class Solution {
private:
    int get_row(int label) {
        int row = -1;
        while (label) {
            row += 1;
            label /= 2;
        }
        return row;
    }
public:
    vector<int> pathInZigZagTree(int label) {
        int row = get_row(label);
        vector<int> res{label};
        while (row) {
            int N = int(std::pow(2, row + 1)) - 1 + int(std::pow(2, row));
            int cur = res.back();
            int prev = (N - cur) / 2;
            res.push_back(prev);
            row --;
        }
        std::reverse(res.begin(), res.end());
        return res;
    }
};
Published 351 original articles · won praise 7 · views 10000 +

Guess you like

Origin blog.csdn.net/Eric_1993/article/details/104580062