二叉树的最大深度

题干

给定一个二叉树,找出其最大深度。

二叉树的深度为根节点到最远叶子节点的最长路径上的节点数。

示例:

给定二叉树 [3,9,20,null,null,15,7],

    3
   / \
  9  20
    /  \
   15   7

返回它的最大深度 3 。

题解

方法一:DFS,递归

public static int maxDepth(TreeNode root) {
	if (root == null)
		return 0;
	int left_height = maxDepth(root.left);
	int right_height = maxDepth(root.right);
	//返回值为root节点左右分支长度较长的那一个
	return java.lang.Math.max(left_height, right_height) + 1;
}

复杂度分析

  • 时间复杂度:我们每个结点只访问一次,因此时间复杂度为 O ( N ) O(N) O(N)
    其中 N N N 是结点的数量。
  • 空间复杂度:在最糟糕的情况下,树是完全不平衡的,例如每个结点只剩下左子结点,递归将会被调用 N N N 次(树的高度),因此保持调用栈的存储将是 O ( N ) O(N) O(N)。但在最好的情况下(树是完全平衡的),树的高度将是 l o g ( N ) log(N) log(N)。因此,在这种情况下的空间复杂度将是 O ( l o g ( N ) ) O(log(N)) O(log(N))

方法二:DFS,迭代
我们从包含根结点且相应深度为 1 的栈开始。然后我们继续迭代:将当前结点弹出栈并推入子结点。每一步都会更新深度。

力扣上官方题解把栈写成了队列,应该是:

import javafx.util.Pair;
public static int maxDepth(TreeNode root) {
	Stack<Pair<TreeNode, Integer>> stack = new Stack<Pair<TreeNode, Integer>>();
        if (root != null) {
            stack.add(new Pair(root, 1));
        }

        int depth = 0;
        while (!stack.isEmpty()) {
            Pair<TreeNode, Integer> current = stack.pop();
            root = current.getKey();
            int current_depth = current.getValue();
            if (root != null) {
                depth = Math.max(depth, current_depth);
                stack.add(new Pair(root.left, current_depth + 1));
                stack.add(new Pair(root.right, current_depth + 1));
            }
        }
        return depth;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值