LeetCode 49. 二叉树的右视图

题目描述: 给定一个二叉树的根节点 root, 想象自己站在它的右侧,按照从顶部到底部的顺序,返回从右侧所能看到的节点值。
示例 1:
LeetCode 49. 二叉树的右视图

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

示例 2:

输入: [1,null,3]
输出: [1,3]

示例 3:

输入: []
输出: []

提示:

  • 二叉树的节点个数的范围是 [0,100]
  • -100 <= Node.val <= 100

来源: 力扣(LeetCode)
链接:https://leetcode-cn.com/problems/binary-tree-right-side-view/
解决方案:
方法一:使用层序遍历的方式,但是只将最层最后一个结点的值加入到结果列表中。

public List<Integer> rightSideView(TreeNode root) {
    List<Integer> res = new ArrayList<>();
    if (root == null) return res;
    Queue<TreeNode> dq = new LinkedList<>();
    dq.offer(root);
    while (!dq.isEmpty()) {
        int size = dq.size();
        while (size > 0) {
            TreeNode node = dq.poll();
            if (node.left != null) dq.offer(node.left);
            if (node.right != null) dq.offer(node.right);
            if (size == 1) res.add(node.val);
            size--;
        }
    }
    return res;
}

复杂度分析:

  • 时间复杂度 : 每个节点最多进队列一次,出队列一次,时间复杂度为 O(n).
  • 空间复杂度 : 每个节点最多进队列一次,所以队列长度最大不超过 n, 所以这里的空间代价为 O(n).

方法二:使用深度优先搜索方式。这里有个注意的点就是,以前我们使用深度优先方式时,先遍历左子树再遍历右子树。在这里是先遍历右子树再遍历左子树,这样在每层见到的第一个结点一定是最右边的结点。

List<Integer> res = new ArrayList<>();
public List<Integer> rightSideView(TreeNode root) {
    dfs(root, 0);
    return res;
}
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);
}

复杂度分析:

  • 时间复杂度 : 深度优先搜索最多访问每个结点一次,因此是线性复杂度,O(n).
  • 空间复杂度 : 最坏情况下,栈内会包含接近树高度的结点数量,占用 O(n) 的空间。
上一篇:GD32系列总结 - systick介绍及使用


下一篇:2021/10/13 智能家具 嵌入式实训 第三天 编写Systick代码 (5)