Tree Boundary Traversal
JavaView on GFG
Time: O(n)
Space: O(h)
Advertisement
Intuition
Three parts: left boundary (excluding leaves), all leaves (left to right), right boundary (excluding leaves, bottom to top).
Algorithm
- 1Left boundary: traverse left spine, add non-leaf nodes.
- 2Leaves: DFS in-order, add leaf nodes.
- 3Right boundary: traverse right spine, add non-leaf nodes in reverse.
Common Pitfalls
- •Avoid double-counting root (if leaf) or leaf nodes in boundary traversal. Reverse right boundary before adding.
Tree Boundary Traversal.java
Java
// Approach: Three parts: left boundary (top-down), leaf nodes (left to right), right boundary (bottom-up).
// Time: O(n) Space: O(h)
class Node {
int data;
Node left, right;
public Node(int d) {
data = d;
left = right = null;
}
}
class Solution {
ArrayList<Integer> boundaryTraversal(Node node) {
ArrayList<Integer> ans = new ArrayList<>();
if (node == null)
return null;
if (!leaf(node))
ans.add(node.data);
left(node, ans);
addleaves(node, ans);
right(node, ans);
return ans;
}
public static boolean leaf(Node node) {
return (node.left == null && node.right == null);
}
void left(Node root, ArrayList<Integer> ans) {
Node curr = root.left;
while (curr != null) {
if (!leaf(curr))
ans.add(curr.data);
if (curr.left != null)
curr = curr.left;
else
curr = curr.right;
}
}
void right(Node root, ArrayList<Integer> ans) {
Node curr = root.right;
Stack<Integer> st = new Stack<>();
while (curr != null) {
if (!leaf(curr))
st.push(curr.data);
if (curr.right != null)
curr = curr.right;
else
curr = curr.left;
}
while (!st.isEmpty())
ans.add(st.pop());
}
void addleaves(Node root, ArrayList<Integer> ans) {
if (root == null)
return;
if (leaf(root))
ans.add(root.data);
else {
addleaves(root.left, ans);
addleaves(root.right, ans);
}
}
}Advertisement
Was this solution helpful?