算法系列——二叉树的右视图

Posted 2022-04-12 BridgeGeorge

题目

给定一个二叉树的 根节点 root，想象自己站在它的右侧，按照从顶部到底部的顺序，返回从右侧所能看到的节点值。

思路

广度优先遍历 输出每一层的最后一个结点即可。

代码

/**
 * Definition for a binary tree node.
 * public class TreeNode 
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode() 
 *     TreeNode(int val)  this.val = val; 
 *     TreeNode(int val, TreeNode left, TreeNode right) 
 *         this.val = val;
 *         this.left = left;
 *         this.right = right;
 *     
 * 
 */
class Solution 
    public List<Integer> rightSideView(TreeNode root) 
        List<Integer> res=new ArrayList<>();
        if(root==null)
            return res;
        
        Queue<TreeNode> queue=new LinkedList<>();
        queue.add(root);
        while(queue.size()>0)
           int size= queue.size();
           for(int i=0;i<size;i++)
              TreeNode node= queue.poll();
                if(node.left!=null)
                    queue.add(node.left);
                
                if(node.right!=null)
                    queue.add(node.right); 
                
                if(i==size-1)
                    res.add(node.val);
                
           
            
        
        return res;