最大深度问题

二叉树的最大深度

二叉树深度和高度的区别

对于某个节点来说:

  • 深度是指从根节点到该节点的最长简单路径边的条数;
  • 高度是指从最下面叶子节点到该节点的最长简单路径边的条数;

对于二叉树来说:

  • 深度是从根节点数到它的叶节点;
  • 高度是从叶节点数到它的根节点;

注意: 树的深度和高度一样,但是具体到树的某个节点,其深度和高度不一样。

题目描述

力扣链接🔗

最大深度问题 - 图1

递归法

后序遍历

求的根节点的高度,而根节点的高度就是这棵树的最大深度,所以才可以使用后序遍历。

  1. 确定递归函数的参数和返回值
    参数就是传入树的根节点,返回就返回这棵树的深度,所以返回值为int类型。
  2. 确定中止条件
    如果为空节点的话,就返回0,表示高度为0。
  3. 确认单层递归的逻辑
    先求左孩子的深度,再求右孩子的深度,最后左右孩子中最大的深度再加上1(此节点)为此节点的最大深度。
  1. /**
  2. * 递归法(后序)
  3. */
  4. public int maxDepth(TreeNode root) {
  5. return getDepth(root);
  6. }
  7. public int getDepth(TreeNode root) {
  8. if (root == null) {
  9. return 0;
  10. }
  11. int leftDepth = getDepth(root.left);
  12. int rightDepth = getDepth(root.right);
  13. return 1 + Math.max(leftDepth, rightDepth); // 返回左右孩子最大的深度 + 本身节点
  14. }

前序遍历

前序遍历是以深度的角度来求,所以没在一个节点都需要更新新的深度。

  1. /**
  2. * 递归法(前序)
  3. *
  4. * @param root
  5. * @return
  6. */
  7. int result = 0;
  8. public int maxDepth(TreeNode root) {
  9. if (root == null) {
  10. return result;
  11. }
  12. getDepth(root, 1);
  13. return result;
  14. }
  15. public void getDepth(TreeNode root, int depth) {
  16. result = Math.max(result, depth); // 此时处理为中节点,将最大的深度作为结果
  17. if (root.left == null && root.right == null) return; // 左右都为空,不需要处理左右孩子
  18. if (root.left != null) {
  19. depth++;
  20. getDepth(root.left, depth); // 左
  21. depth--; // 到达最后一个节点,回溯
  22. }
  23. if (root.right != null) {
  24. depth++;
  25. getDepth(root.right, depth); // 右
  26. depth--; // 到达最后一个节点,回溯
  27. }
  28. return;
  29. }

迭代遍历

使用层序遍历模板,遍历一层加一即可。

最大深度问题 - 图2

  1. /**
  2. * 使用层序遍历
  3. */
  4. public int maxDepth(TreeNode root) {
  5. Queue<TreeNode> queue = new LinkedList<>();
  6. int deep = 0; // 深度
  7. if (root == null) {
  8. return 0;
  9. }
  10. queue.offer(root);
  11. while (!queue.isEmpty()) {
  12. int len = queue.size();
  13. while (len > 0) {
  14. TreeNode node = queue.poll();
  15. if (node.left != null) queue.offer(node.left);
  16. if (node.right != null) queue.offer(node.right);
  17. len--;
  18. }
  19. deep++; // 深度加一
  20. }
  21. return deep;
  22. }

N叉数的最大深度

最大深度问题 - 图3

与二叉树的最大深度思路一致

递归法

同样是求出子节点的最大深度,再加上一,为当前节点的最大深度,在进行递归。

  1. /**
  2. * 递归遍历(后序)
  3. *
  4. * @param root
  5. * @return
  6. */
  7. public int maxDepth(Node root) {
  8. if (root == null) {
  9. return 0;
  10. }
  11. int depth = 0;
  12. for (int i = 0; i < root.children.size(); i++) {
  13. depth = Math.max(depth, maxDepth(root.children.get(i))); // 递归求出此节点的最大深度
  14. }
  15. return depth + 1;
  16. }

迭代法

  1. /**
  2. * 层序遍历
  3. *
  4. * @param root
  5. * @return
  6. */
  7. public int maxDepth(Node root) {
  8. if (root == null) {
  9. return 0;
  10. }
  11. Queue<Node> queue = new LinkedList<>();
  12. int depth = 0;
  13. queue.offer(root);
  14. while (!queue.isEmpty()) {
  15. int len = queue.size();
  16. while (len > 0) {
  17. Node node = queue.poll();
  18. for (Node cilNode : node.children) {
  19. if (cilNode != null) queue.offer(cilNode);
  20. }
  21. len--;
  22. }
  23. depth++;
  24. }
  25. return depth;
  26. }