标题
输入一棵二叉树的根节点,求该树的深度。从根节点到叶节点依次经过的节点(含根、叶节点)形成树的一条路径,最长路径的长度为树的深度。
解题思路
可以用递归的方法实现,我们假设已经正确拿到了root节点左右子树的最大深度,那么最后加1即可:
- 如果一棵树只有一个节点,那么它的深度为1;
- 如果根节点只有左子树而没有右子树,那么树的深度应该是其左子树的深度加1;
- 如果根节点只有右子树而没有左子树,那么树的深度应该是其右子树的深度加1;
- 如果既有右子树又有左子树,那么该树的深度就是其左、右子树深度的较大值再加1。
代码
/*
struct TreeNode {
int val;
struct TreeNode *left;
struct TreeNode *right;
TreeNode(int x) :
val(x), left(NULL), right(NULL) {
}
};*/
class Solution{
public:
int TreeDepth(TreeNode* pRoot){
//递归的出口,root为0则返回0,这里可以理解为root为0那肯定没有层数了
if(pRoot == nullptr) return 0;
//拿到左子树的最大深度
int left = TreeDepth(pRoot->left);
//拿到右子树的最大深度
int right = TreeDepth(pRoot->right);
//找出最大值,并且加上root这一层即可
return (left > right) ? (left +1):(right +1);
}
};