题目描述

求给定二叉树的最大深度,

最大深度是指树的根结点到最远叶子结点的最长路径上结点的数量。

方法一: 递归

解题思路

二叉树的问题最常用的方法就是递归,将大规模的问题划成小规模的问题,每个问题的求解方法都是一样的,最后规模小到方便求出答案,再由小规模的问题答案求出大规模的问题答案。

求以根节点为二叉树的最大深度,可以先求出根节点的左子树上的高度是多少,再求出右子树上的高度是多少,这两者的最大高度就是以根节点为二叉树的最大深度。

根节点的左子树,仍然是一棵二叉树,也可以用同样的方法求出它的最大深度。

根节点的右子树,也仍然是一颗二叉树,也可以用同样的方法求出它的最大深度。


以下是以 [1,0,2,8] 为二叉树的 流程演示:

图片说明

复杂度分析

时间复杂度: O(N)

每个节点都会被递归遍历访问到,一共有 N 个节点, 所以时间复杂度为 O(N)

空间复杂度: O(N)

空间上的开销主要在 递归栈 的存储上,最多的递归栈就是这棵二叉树的高度个,即O(h),又因为二叉树的高度 h = logN,所以空间复杂度为 O(logN);
但最坏的情况下,可以把每个节点都压入栈中,即空间复杂度可以达到 O(N)。

代码示例:

import java.util.*;

/*
 * public class TreeNode {
 *   int val = 0;
 *   TreeNode left = null;
 *   TreeNode right = null;
 * }
 */

public class Solution {
    /**
     * 
     * @param root TreeNode类 
     * @return int整型
     */
    public int maxDepth (TreeNode root) {
        if (root == null) {
            return 0;
        }

        // 整颗树的根节点所处的高度为 1
        return process(root, 1);
    }

    /**
     * 
     * @param root  树的根节点
     * @param depth 这个根节点所处 的高度
     * @return 以 root 为根节点的树 的高度是多少
     */
    public int process(TreeNode root, int depth) {
        // 当 root 节点不存在, 则高度就是它的父节点所处的高度
        // 否则就是 root 节点本身所处的高度,因为不存在,所以不会是 depth,而是它上一级的高度
        if (root == null) {
            return depth - 1;
        }

        // root 节点的左子树高度
        int leftDepth = process(root.left, depth + 1);
        // root 节点的右子树高度
        int rightDepth = process(root.right, depth + 1);

        // 左右子树较高的那个高度,就是整颗树的高度
        return Math.max(leftDepth, rightDepth);
    }
}

方法二: 层序遍历

解题思路

二叉树的层序遍历可以很自然的求出树的高度,有多少层它的高度就是多少。

实现层序遍历最常用的方法就是借助队列结构,先将一层上的节点装入队列中,然后再从队列中以此拿出来,看看每个节点是否有左右孩子,也就是看看每个节点是否有下一层,如果有,就将下一层的节点再次入队。

需要注意的是每次遍历完一层之后,高度就会增加1,最简单的方法就是需要知道每层有多少个节点,拿出指定个数的节点出来就是遍历完了一层。

具体思路请参见示例代码。


以下是以 [1,0,2,8] 为二叉树的 流程演示:

图片说明

复杂度分析

时间复杂度: O(N)

每个节点都需要遍历到,总共有 N 个节点,所以时间复杂度为 O(N)

空间复杂度:O(N)

空间上的开销主要是在队列的存储上,最多的时候就是看哪一层的节点个数最多,

极端情况下就是一颗满二叉树,叶子节点那一层是最多的,所以空间复杂度为 O(N)

代码示例

import java.util.*;

/*
 * public class TreeNode {
 *   int val = 0;
 *   TreeNode left = null;
 *   TreeNode right = null;
 * }
 */

public class Solution {
    /**
     * 
     * @param root TreeNode类 
     * @return int整型
     */
    public int maxDepth(TreeNode root) {
        if (root == null) {
            return 0;
        }

        // 层序遍历需要用到的 队列
        Queue<TreeNode> queue = new LinkedList<>();
        // 将个根节点入队
        queue.add(root);
        // 记录高度的信息,初始为 0, 每遍历一层就加 1
        int depth = 0;

        // 当队列不为空时,也就是这一层还有节点
        while (!queue.isEmpty()) {
            // 这一层有多少个节点
            int size = queue.size();
            // 挨个取出这些(这一层)节点
            for (int i = 0; i < size; i++) {
                TreeNode node = queue.poll();
                // 当前节点有 左节点, 使左节点入队
                if (node.left != null) {
                    queue.add(node.left);
                }
                // 当前节点有 右节点, 使右节点入队
                if (node.right != null) {
                    queue.add(node.right);
                }
            }
            // 每遍历一层, 高度就加 1
            depth++;
        }

        // 返回整颗树的高度
        return depth;
    }

}