二叉树的深度
输入一棵二叉树的根节点,求该树的深度。从根节点到叶节点依次经过的节点(含根、叶节点)形成树的一条路径,最长路径的长度为树的深度。
例如:
给定二叉树 [3,9,20,null,null,15,7],
3
/ \
9 20
/ \
15 7
返回它的最大深度 3 。
提示:
- 节点总数 <= 10000
解法:根节点的深度等于左右子树深度的最大值加一。可以使用递归法实现这一思路。
还可以使用之前的层遍历,通过记录层信息,来计算树的深度。
递归法:
class Solution {
public:
int maxDepth(TreeNode* root) {
if(root == nullptr)
return 0;
int leftDepth = maxDepth(root->left);
int rightDepth = maxDepth(root->right);
return max(leftDepth,rightDepth) + 1;
}
};
层遍历的非递归方法:
class Solution {
public:
int maxDepth(TreeNode* root) {
if(root == nullptr)
return 0;
queue<TreeNode*> queue1;
int nextLevelCount = 0; // 下一层要打印的数量
int toBePrintCount = 1; // 本层将要打印的数量
int depth = 0;
queue1.push(root);
while(!queue1.empty()){
while(toBePrintCount-- > 0){
if(queue1.front()->left != nullptr){
queue1.push(queue1.front()->left);
nextLevelCount++;
}
if(queue1.front()->right != nullptr){
queue1.push(queue1.front()->right);
nextLevelCount++;
}
queue1.pop();
}
depth++; // 打印完一层计数一次
toBePrintCount = nextLevelCount;
nextLevelCount = 0;
}
return depth;
}
};