力扣Java版个人代码分享-树篇(199. 二叉树的右视图)

199. 二叉树的右视图

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

例子

输入: [1,2,3,null,5,null,4]
输出: [1, 3, 4]

解释:

   1            <---
 /   \
2     3         <---
 \     \
  5     4       <---

代码

   public List<Integer> rightSideView(TreeNode root) {
        List<Integer> l = new LinkedList<>();
        Queue<TreeNode> q = new LinkedList<>();
        if(root == null){
            return l;
        }
        q.add(root);
        while(!q.isEmpty()){
            int i = q.size();
            TreeNode temp = null;
            while(i > 0){
                --i;
                temp = q.peek();
                if(temp.left != null){
                    q.add(temp.left);
                }
                if(temp.right != null){
                    q.add(temp.right);
                }
                q.poll();
            }
            l.add(temp.val);
        }
        return l;
    }

思路

使用层序遍历，取出每一层最右元素

注意事项

待优化