LeetCode.0199二叉树的右视图

0199.二叉树的右视图

描述

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

实例

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

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

题解

逐层遍历,输出每次List的最后一个元素.

public ArrayList<TreeNode> getNextLayNodes(ArrayList<TreeNode> layList1){
        ArrayList<TreeNode> layList2 = new ArrayList<>();
        for(TreeNode node:layList1){
            if (node.left != null)
                layList2.add(node.left);
            if (node.right != null)
                layList2.add(node.right);
        }
        return layList2;
    }

    public List<Integer> rightSideView(TreeNode root) {
        List<Integer> result = new ArrayList<>();
        if (root == null)
            return result;
        ArrayList<TreeNode> layList1;
        ArrayList<TreeNode> layList2;
        layList1 = new ArrayList<>();
        layList1.add(root);

        while (!layList1.isEmpty()){
            layList2 = getNextLayNodes(layList1);
            result.add(layList1.get(layList1.size()-1).val);
            layList1 = layList2;
        }

        return result;
    }