Leetcode 199 题 二叉树的右视图

题目描述

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

示例：
输入：[1,2,3,null,5,null,4]
输出：[1, 3, 4]
解释：

思路分析

显然，我们需要获得每层最右边的元素，加入到结果集中。

「广度优先搜索」

层序遍历，按照从右到左的顺序访问节点，每层第一次被访问的元素即为最右边元素。

class Solution {
    public List<Integer> rightSideView(TreeNode root) {
        List<Integer> result = new ArrayList<>();
        if (root == null) return result;

        Queue<TreeNode> queue = new LinkedList<>();
        queue.add(root);

        while (!queue.isEmpty()) {
            int size = queue.size();                // 记录当前层中节点个数
            for (int i = 0; i < size; i++) {        // 遍历当前层元素
                TreeNode curr = queue.remove();     // 取出队首元素
                if (i == 0) result.add(curr.val);   // 当前层最右边的元素
                if (curr.right != null) queue.add(curr.right);
                if (curr.left != null) queue.add(curr.left);
            }
        }

        return result;
    }
}

「深度优先搜索」

这里先遍历右子树，再遍历左子树，每层最右边的节点一定最先被访问。

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

    private void dfs(TreeNode node, int level) {
        if (node == null) return;

        if (level == result.size()) {
            result.add(node.val);
        }

        if (node.right != null) dfs(node.right, level + 1);
        if (node.left != null) dfs(node.left, level + 1);
    }
}

技术公众号：小猿君的算法笔记