BFS 01

199. 二叉树的右视图

原题链接

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    public List<Integer> rightSideView(TreeNode root) {
        List<Integer> res = new ArrayList<>();          //创建结果集合
        if(root == null) {
            return res;
        }
        Queue<TreeNode> queue = new LinkedList<>();     //创建队列维护
        queue.offer(root);
        while(!queue.isEmpty()) {
            int size = queue.size();
            for(int i = 0; i < size; i++ ) {            //遍历当前层的所有节点
                TreeNode node = queue.poll();           //返回头部元素
                if(node.left != null) {
                    queue.offer(node.left);             
                }                                       //将当前节点的左右孩子入队
                if(node.right != null) {
                    queue.offer(node.right);           
                }
                if(i == size - 1) {            
                    res.add(node.val);                  //将当前层的最后一个节点放入
                }
            }
        }
        return res;
    }
}