算法---LeetCode 199. 二叉树的右视图

1. 题目

原题链接
给定一棵二叉树，想象自己站在它的右侧，按照从顶部到底部的顺序，返回从右侧所能看到的节点值。
示例:
输入: [1,2,3,null,5,null,4]
输出: [1, 3, 4]
解释:
1 <—
/
2 3 <—
\
5 4 <—

Related Topics 树 深度优先搜索 广度优先搜索 递归 队列
👍 479 👎 0

2. 题解

2.1 解法1: BFS 层次遍历

记录每层最后一个结点

    class Solution {
        public List<Integer> rightSideView(TreeNode root) {
            List<Integer> ans = new ArrayList<>();
            if (root == null) {
                return ans;
            }
            Queue<TreeNode> queue = new LinkedList<>();
            queue.offer(root);
            while (!queue.isEmpty()) {
                int size = queue.size();
                for (int i = 0; i < size; i++) {
                    TreeNode temp = queue.poll();
                    if (temp.left != null) {
                        queue.offer(temp.left);
                    }
                    if (temp.right != null) {
                        queue.offer(temp.right);
                    }
                    if (i == size - 1) {
                        ans.add(temp.val);
                    }
                }
            }
            return ans;
        }
    }

2.2 解法2: DFS 递归

以根, 右, 左, 顺序遍历该树, 那么每次遍历到该层的第一个结点一定是最右边的结点,原因可以参考先序遍历, 在每一层首次访问的结点是最左结点

在实现时, 可以使用一个depth 变量记录在哪一层, 并在递归中通过结果集长度比较, 判断是否是该层第一个元素

    class Solution {
        List<Integer> ans;

        public List<Integer> rightSideView(TreeNode root) {
            ans = new ArrayList<>();
            dfs(root, 0);
            return ans;
        }

        public void dfs(TreeNode root, int depth) {
            if (root == null) {
                return;
            }
            if (depth == ans.size()) {
                ans.add(root.val);
            }
            depth++;
            dfs(root.right, depth);
            dfs(root.left, depth);
        }

    }

参考:
简单BFS/DFS, 🤷‍♀️必须秒懂！