题目:
输入一棵二叉树的根节点,求该树的深度。从根节点到叶节点依次经过的节点(含根、叶节点)形成树的一条路径,最长路径的长度为树的深度。
例如:
给定二叉树 [3,9,20,null,null,15,7],
返回它的最大深度 3 。
树的遍历方式总体分为两类:深度优先搜索(DFS)和广度优先搜索(BFS)
常见的DFS:先序遍历、中序遍历、后序遍历
常见的BFS:层序遍历
求树的深度需要遍历树的所有节点,本文将介绍基于后序遍历DFS和层序遍历BFS
方法一:后序遍历DFS
树的后序遍历/DFS往往采用递归或者栈实现
关键点,此树的深度和其左右子树深度的关系,可知,树的深度等于根节点的左子树和右子树深度的最大值+1
使用递归,那么需要终止条件,显然在root==null的时候终止返回0
算法实现:
java
/**
* 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) {
return root==null?0:Math.max(maxDepth(root.left),maxDepth(root.right))+1;
}
}
python
# Definition for a binary tree node.
# class TreeNode:
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class Solution:
def maxDepth(self, root: TreeNode) -> int:
if not root:
return 0
return max(self.maxDepth(root.left), self.maxDepth(root.right)) + 1
方法二、层序遍历
树的层序遍历 / 广度优先搜索往往利用 队列 实现。
关键点: 每遍历一层,则计数器 +1 ,直到遍历完成,则可得到树的深度。
算法解析:
- 当root为空的时候,直接return 0
- root不为空的时候,插入队列中,此时根节点位于队列头,计数器为0
- size为当前队列中的节点数,也即处于当前层的树有对应几个节点
- 然后以size为遍历终点,往队列中插入这几个节点的子节点也即下一层的树节点。
- 循环遍历直到队列为空,也就是所有节点都遍历完毕,此时计数器中的值就是最大深度
/**
* 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) {
Queue<TreeNode> queue = new LinkedList<>();
TreeNode temp = root;
if (root==null){
return 0;
}
queue.offer(temp);
int result=0;
while (!queue.isEmpty()){
int size = queue.size();
for(int i=0;i<size;i++){
TreeNode tempNode = queue.poll();
if (tempNode.left!=null){
queue.offer(tempNode.left);
}
if (tempNode.right!=null){
queue.offer(tempNode.right);
}
}
result+=1;
}
return result;
}
}