知识点:二叉树,递归,二叉树的遍历
题目来源:nc274
输入一棵二叉树,求该树的深度。从根结点到叶结点依次经过的结点(含根、叶结点)形成树的一条路径,最长路径的长度为树的深度,根节点的深度视为 1 。
数据范围:节点的数量满足 0≤n≤1000 ,节点上的值满足 0≤val≤1000
空间复杂度 O(1) ,时间复杂度 O(n)
假如输入的用例为{1,2,3,4,5,#,6,#,#,7},那么如下图:
测试数据
示例1
输入:
{1,2,3,4,5,#,6,#,#,7}
返回值:
4
示例2
输入:
{}
返回值:
0
算法思路:
通常看到与二叉树有关的题,最常用的几种解法分别为,1)前序遍历;2)中序遍历;3)后序遍历;4)层次遍历(类比图的dfs,bfs);5)递归解法。
显然,本题可以使用,
1)层次遍历,即通过一个队列,每遍历完一层,则层数+1,遍历完成后返回层数即可(但是层次遍历模板比较繁琐,所以我这里没用,而是采用下面的方法);
2)递归解法,即分别递归求解左子树,右子树的深度,其中的最大值+1,即为本层的深度,递归返回时最后的深度即为整颗二叉树的最大深度。
代码实现:
//结点定义
struct TreeNode {
int val;
struct TreeNode *left;
struct TreeNode *right;
};
//代码实现
int TreeDepth(TreeNode* pRoot) {
if(!pRoot) return 0;
int a = TreeDepth(pRoot->left)+1;
int b = TreeDepth(pRoot->right)+1;
return a>b? a:b;
}