Welcome to Subscribe On Youtube
Formatted question description: https://leetcode.ca/all/1104.html
1104. Path In Zigzag Labelled Binary Tree (Medium)
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.
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
Solution 1.
// OJ: https://leetcode.com/problems/path-in-zigzag-labelled-binary-tree/
// Time: O(logN)
// Space: O(1)
class Solution {
inline int r(int label, int row) {
return 3 * (1 << row) - label - 1;
}
public:
vector<int> pathInZigZagTree(int label) {
vector<int> ans(1, label);
while (label != 1) {
int row = log2(label);
if (row % 2) label = r(label, row) / 2;
else label = r(label / 2, row - 1);
ans.push_back(label);
--row;
}
reverse(ans.begin(), ans.end());
return ans;
}
};
-
class Solution { public List<Integer> pathInZigZagTree(int label) { Stack<Integer> stack = new Stack<Integer>(); while (label >= 1) { stack.push(label); int log = (int) (Math.log(label) / Math.log(2)); int minValue = (int) Math.pow(2, log); label = minValue * 3 - 1 - label; label /= 2; } List<Integer> path = new ArrayList<Integer>(); while (!stack.isEmpty()) path.add(stack.pop()); return path; } } -
// OJ: https://leetcode.com/problems/path-in-zigzag-labelled-binary-tree/ // Time: O(logN) // Space: O(1) class Solution { inline int r(int label, int row) { return 3 * (1 << row) - label - 1; } public: vector<int> pathInZigZagTree(int label) { vector<int> ans(1, label); while (label != 1) { int row = log2(label); if (row % 2) label = r(label, row) / 2; else label = r(label / 2, row - 1); ans.push_back(label); --row; } reverse(ans.begin(), ans.end()); return ans; } }; -
# 1104. Path In Zigzag Labelled Binary Tree # https://leetcode.com/problems/path-in-zigzag-labelled-binary-tree/ class Solution: def pathInZigZagTree(self, label: int) -> List[int]: res = [] level = nodes = 1 while label >= nodes * 2: nodes *= 2 level += 1 while label != 0: res.append(label) mmax = (2 ** level) - 1 mmin = 2 ** (level - 1) label = (mmax + mmin - label) // 2 level -= 1 return res[::-1]