LeetCode 104. Maximum Depth of Binary Tree 二叉树的最大深度

题目:

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

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

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

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

    3
   / \
  9  20
    /  \
   15   7

返回它的最大深度 3 。

解题思路:

通过递归(前序遍历)或者非递归(中序遍历)的方式获得二叉树的最大深度。非递归方法可以体会一下使用两个stack的妙用

代码实现:

递归版本:

/**
 * 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;
        return Integer.max(maxDepth(root.left), maxDepth(root.right)) + 1;
    }
}

非递归版本:

/**
 * 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;
        Stack<TreeNode> nodes = new Stack<TreeNode>();
        Stack<Integer> values = new Stack<Integer>();
        TreeNode p = root;
        int height = 0;
        int max = 0;
        while (p != null) {
            nodes.push(p);
            values.push(++ height);
            p = p.left;
            while (p == null && !nodes.isEmpty()) {
                p = nodes.pop();
                height = values.pop();
                max = Integer.max(max, height);
                p = p.right;
            }
        }
        return max;
    }
}

猜你喜欢

转载自blog.csdn.net/zhangzhetaojj/article/details/80704752