[GeeksForGeeks] Print leftmost and rightmost nodes at each level of a binary tree.

Given a Binary Tree, Print the corner nodes at each level. The node at the leftmost and the node at the rightmost.

For example, output for following is 15, 10, 20, 8, 25.

 

Solution. Level Order Traversal using queue.

Core idea:  Level order traversal always visit nodes of one level from left to right. 

And we know the number of nodes at each level by pre-reading the size of the queue. 

 

import java.util.ArrayList;
import java.util.LinkedList;
import java.util.Queue;

class TreeNode {
    TreeNode left;
    TreeNode right;
    int val;
    TreeNode(int val){
        this.left = null;
        this.right = null;
        this.val = val;
    }
}
public class Solution {
    public ArrayList<TreeNode> getLeftRightMostAtEachLevel(TreeNode root) {
        ArrayList<TreeNode> result = new ArrayList<TreeNode>();
        if(root == null){
            return result;
        }
        Queue<TreeNode> queue = new LinkedList<TreeNode>();
        queue.offer(root);        
        while(!queue.isEmpty()){
            int size = queue.size();
            for(int i = 0; i < size; i++){
                TreeNode curr = queue.poll();
                if(i == 0){
                    result.add(curr);
                }
                if(i > 0 && i == size - 1){
                    result.add(curr);
                }
                if(curr.left != null){
                    queue.offer(curr.left);
                }
                if(curr.right != null){
                    queue.offer(curr.right);
                }
            }
        }
        return result;
    }
}