题目描述

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

样例

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

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

算法1

dfs

• 1、按照 「根结点 -> 右子树 -> 左子树」 的顺序访问，就可以保证每层都是最先访问最右边的节点的。
• 2、若当前深度depth == ans.size()，表示当前深度没有元素记录，则需要记录该深度对应的元素，因此按照1的遍历顺序进行访问，记录下来的元素肯定是当前深度右视图看到的

时间复杂度 $O(n)$

Java 代码

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    static List<Integer> ans = new ArrayList<Integer>();
    static void dfs(TreeNode root, int depth)
    {
        if(root == null)
            return ;

        if(depth == ans.size())
            ans.add(root.val);

        dfs(root.right, depth + 1);
        dfs(root.left, depth + 1);
    }
    public List<Integer> rightSideView(TreeNode root) {
        ans.clear();
        if(root == null) return ans;
        dfs(root, 0);
        return ans;
    }
}

算法2

bfs

• 利用 bfs 进行层次遍历，右视图看到的元素则是当前层最后一个元素

时间复杂度 $O(n)$

Java 代码

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    static List<Integer> ans = new ArrayList<Integer>();
    //记录每一层的最后一个元素
    static void bfs(TreeNode root)
    {
        Queue<TreeNode> q = new LinkedList<TreeNode>();
        q.add(root);
        while(!q.isEmpty())
        {
            int n = q.size();
            for(int i = 0;i < n ;i ++) 
            {
                TreeNode t = q.poll();
                if(i == n - 1) ans.add(t.val);
                if(t.left != null) q.add(t.left);
                if(t.right != null) q.add(t.right);
            }
        }
    }
    public List<Integer> rightSideView(TreeNode root) {
        ans.clear();
        if(root == null) return ans;
        bfs(root);
        return ans;
    }
}