二叉樹的右檢視 (中等)

題目

給定一棵二叉樹，想象自己站在它的右側，按照從頂部到底部的順序，返回從右側所能看到的節點值。

示例:

輸入: [1,2,3,null,5,null,4]
輸出: [1, 3, 4]
解釋:

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

解題思路

1. 儲存每一層的最又邊的值

程式碼

/**
 * 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) {
        if(root == null) return new ArrayList<>();
        ArrayList<TreeNode> nodes = new ArrayList<>();
        nodes.add(root);
        ArrayList<Integer> vals = new ArrayList<>();
        while (!nodes.isEmpty()){
            ArrayList<TreeNode> tmp = new ArrayList<>();
            vals.add(nodes.get(nodes.size()-1).val);
            for (TreeNode node : nodes){
                if(node.left != null){
                    tmp.add(node.left);
                }
                if(node.right != null){
                    tmp.add(node.right);
                }
            }
            nodes = tmp;
        }
        return vals;
    }
}

本作品採用《CC 協議》，轉載必須註明作者和本文連結