[剑指offer]二叉树的深度
题目描述
输入一棵二叉树的根节点,求该树的深度。从根节点到叶节点依次经过的节点(含根、叶节点)形成树的一条路径,最长路径的长度为树的深度。
例如:
给定二叉树 [3,9,20,null,null,15,7],
3
/ \
9 20
/ \
15 7
返回它的最大深度 3 。
提示:
节点总数 <= 10000
解题思路
-
树的遍历总体分两类:深度优先遍历(DFS)和广度优先遍历(BFS)
- 常见的DFS:先序遍历、中序遍历、后序遍历
- 常见的BFS:层序遍历
-
求树的深度需要遍历树的所有节点
-
方法一:后序遍历(DFS)
树的后序遍历一般利用递归和栈实现,本文使用递归实现。
关键点: 树的深度和其左右子树的深度之间的关系。显然,树的深度等于左子树的深度 与 右子树的深度中的 最大值 +1
算法解析:
终止条件:root为空,说明其已经越过叶子节点,返回深度为0;
递推工作:本质是做后序遍历,计算左子树和右子树的深度;
返回值:返回 max(左子树的深度,右子树的深度) + 1,即树的深度。 -
方法二:层序遍历(BFS)
树的层序遍历一般使用队列实现
关键点: 每遍历一层,计数器加1,直到遍历完成,则可得到树的深度
算法解析:
特例:root为空,直接返回深度0;
初始化:根节点rooe入队que,计数器res=0;
循环:que为空时跳出,初始化一个tmp临时存储下一层节点,遍历que的节点node,将其左右节点入队tmp,更新que为tmp,计数器res加1;
返回值:返回计数器res。
实现代码
//递归(后序遍历)
class Solution {
public:
int maxDepth(TreeNode* root) {
return helper(root);
}
int helper(TreeNode* root){
if(root==NULL)
return 0;
return max(helper(root->left),helper(root->right))+1;
}
};
//非递归(层序遍历)
class Solution {
public:
int maxDepth(TreeNode* root) {
if(root==NULL)
return 0;
int res=0;
queue<TreeNode*> que;
que.push(root);
while(!que.empty()){
queue<TreeNode*> tmp;//临时存储下一层节点
while(!que.empty())//遍历que中的节点
{
TreeNode* node=que.front();
que.pop();
if(node->left)tmp.push(node->left);//其左右节点入队tmp
if(node->right)tmp.push(node->right);
}
que=tmp;//更新que=tmp
res++;//计数器加1
}
return res;
}
};