【LeetCode】二叉树的最大深度

题目解释:

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

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

说明: 叶子节点是指没有子节点的节点。

递归思想计算二叉树的最大深度

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

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    public int maxDepth(TreeNode root) {
        if (root == null) {
            return 0;
        }
        int left = maxDepth(root.left);
        int right = maxDepth(root.right);
        return Math.max(left, right) + 1;
    }
}

非递归实现计算二叉树的最大深度(借助于栈将上面的递归转换为迭代)

我们的想法是使用 DFS 策略访问每个结点,同时在每次访问时更新最大深度。

  • 时间复杂度:O(N)。
  • 空间复杂度:O(N)。
import java.util.*;
import javafx.util.Pair;
/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    public int maxDepth(TreeNode root) {
        if (root == null) {
            return 0;
        }
        Queue> queue = new LinkedList<>();
        queue.add(new Pair(root, 1));
        int depth = 0;
        while (!queue.isEmpty()) {
            Pair pair = queue.poll();
            TreeNode curr = pair.getKey();
            int currDepth = pair.getValue();

            if (curr != null) {
                depth = Math.max(depth, currDepth);
                queue.add(new Pair(curr.left, currDepth + 1));
                queue.add(new Pair(curr.right, currDepth + 1));
            }
        }

        return depth;
    }
}

 

你可能感兴趣的:(LeetCode,二叉树的最大深度)