题目描述
求给定二叉树的最大深度,
最大深度是指树的根结点到最远叶子结点的最长路径上结点的数量。
方法一: 递归
解题思路
二叉树的问题最常用的方法就是递归,将大规模的问题划成小规模的问题,每个问题的求解方法都是一样的,最后规模小到方便求出答案,再由小规模的问题答案求出大规模的问题答案。
求以根节点为二叉树的最大深度,可以先求出根节点的左子树上的高度是多少,再求出右子树上的高度是多少,这两者的最大高度就是以根节点为二叉树的最大深度。
根节点的左子树,仍然是一棵二叉树,也可以用同样的方法求出它的最大深度。
根节点的右子树,也仍然是一颗二叉树,也可以用同样的方法求出它的最大深度。
以下是以 [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;
}
} 
京公网安备 11010502036488号