DFS & BFS -leetcode -199. 二叉树的右视图

最新推荐文章于 2024-09-09 22:01:06 发布

pospre

最新推荐文章于 2024-09-09 22:01:06 发布

阅读量111

点赞数

分类专栏： leetcode

本文链接：https://blog.csdn.net/weixin_56915683/article/details/120860883

版权

leetcode 算法二叉树

leetcode 专栏收录该内容

77 篇文章 1 订阅

订阅专栏

199. 二叉树的右视图

题目描述

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

在这里插入图片描述

层次遍历- BFS

按照层析遍历，取每层最后一个元素，“收集” 到 res 中

class Solution {
    public List<Integer> rightSideView(TreeNode root) {
        // List<List<Integer>> res = new ArrayList<>();
        List<Integer> rightView = new ArrayList<>(); // 保存各层的最右节点
        Queue<TreeNode> queue = new LinkedList<>();
        if (root != null) {
            queue.add(root);
        }
        while (!queue.isEmpty()) {
            int size = queue.size();
            // List<Integer> cntLayer = new ArrayList<>(); // 保存当前层所有节点
            for (int i = 0; i < size; i++) {
                TreeNode node = queue.poll();
                if (i == size - 1) { // 收集每层的最后一个元素
                    rightView.add(node.val);
                }
                if (node.left != null) {
                    queue.add(node.left);
                }
                if (node.right != null) {
                    queue.add(node.right);
                }
            }
        }
        return rightView;
    }
}

DFS

dfs 参考题解

思路：

采用 “根右左” 遍历二叉树；
只有第一次在一层出现时，才用 res 收集 🔥

class Solution {
    List<Integer> res = new ArrayList<>();

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

    public void dfs(TreeNode root, int deepth) {
        if (root == null) return;
        // 根
        if (deepth == res.size()) { // 只有第一次在一层出现时，才用 res 收集
            res.add(root.val);
        }
        // res.add(root.val);
        dfs(root.right, deepth + 1); // 右
        dfs(root.left, deepth + 1); // 左
    }
}