题目
给定一个二叉树,找出其最大深度。
二叉树的深度为根节点到最远叶子节点的最长路径上的节点数。
说明: 叶子节点是指没有子节点的节点。
解法1:递归 DFS
对于二叉树的最大深度,可通过递归进行DFS。当节点为空时,直接返回0。若节点不为空,则在左右子树的最大深度值的基础上加1即可。代码如下:
class Solution {
public:
int maxDepth(TreeNode* root) {
if (root == nullptr) {
return 0;
}
return max(maxDepth(root->left), maxDepth(root->right)) + 1;
}
};
时间复杂度为O(n)
,空间复杂度为O(h)
,h为二叉树的高度。执行结果如下:
解法2:迭代 BFS
对于二叉树的最大深度,可通过一个队列辅助进行层序遍历。队列中缓存的数据为同一层的节点,在访问同层节点时,把下一层的节点压入到队列中。如果最后队列为空,则表示该层为最大深度。代码如下:
class Solution {
public:
int maxDepth(TreeNode* root) {
queue<TreeNode*> buffer{};
if (root != nullptr) {
buffer.push(root);
}
int depth = 0;
while (!buffer.empty()) {
size_t count = buffer.size();
for (size_t i = 0; i < count; ++i) {
root = buffer.front();
buffer.pop();
if (root->left != nullptr) {
buffer.push(root->left);
}
if (root->right != nullptr) {
buffer.push(root->right);
}
}
++depth;
}
return depth;
}
};
时间复杂度为O(n)
,空间复杂度为O(n)
。执行结果如下: