199. 二叉树的右视图

题目

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

思路(Switte)

根节点-> 右子树 -> 左子树 顺序访问，保证每层最先访问最右边的节点
时间复杂度：O(n) 每个节点访问了一次
空间复杂度：O(n) 递归用的栈为O(n) 递归栈接近树高度的结点

代码

package binaryTree;
import java.util.ArrayList;
import java.util.List;
public class o199 {
    List<Integer> res = new ArrayList<>();

    public List<Integer> rightSideView(TreeNode root) {
        dfs(root, 0);
        return res;
    }

    private void dfs(TreeNode root,int depth) {
        if (root == null) {
            return;
        }
        if (depth == res.size()) {
            res.add(root.val);
        }
        depth++;
        dfs(root.right,depth);
        dfs(root.left,depth);
    }
    public static void main(String[] args) {
        TreeNode n5 = new TreeNode(5);
        TreeNode n4 = new TreeNode(4);
        TreeNode n2 = new TreeNode(2, null, n5);
        TreeNode n3 = new TreeNode(3, null, n4);
        TreeNode n1 = new TreeNode(1, n2, n3);
        o199 main = new o199();
        System.out.println(main.rightSideView(n1));
    }
}