给定一个二叉树 root
,返回其最大深度。
二叉树的 最大深度 是指从根节点到最远叶子节点的最长路径上的节点数。
示例 1:
输入:root = [3,9,20,null,null,15,7] 输出:3
示例 2:
输入:root = [1,null,2] 输出:2
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val; // 节点存储的整数值
* TreeNode *left; // 指向当前节点的左子节点的指针
* TreeNode *right; // 指向当前节点的右子节点的指针
* TreeNode() : val(0), left(nullptr), right(nullptr) {} // 默认构造函数
* TreeNode(int x) : val(x), left(nullptr), right(nullptr) {} // 带有节点值的构造函数
* TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {} // 完整的构造函数
* };
*/
class Solution {
public:
// 计算二叉树的最大深度
int maxDepth(TreeNode* root) {
// 如果根节点为空,则深度为0
if(root == nullptr)
return 0;
// 递归计算左子树和右子树的最大深度,并取两者中的最大值
// 然后加上根节点本身的深度(+1)
return max(maxDepth(root->left), maxDepth(root->right)) + 1;
}
};
这个类定义了一个二叉树节点 TreeNode
,其包含一个整数值 val
和指向其左右子节点的指针。maxDepth
函数用于计算二叉树的最大深度。它使用递归的方法来计算,递归的基准情况是当根节点为空时,此时深度为0。如果根节点不为空,函数会递归地计算左子树和右子树的最大深度,并将它们中的最大值加1(根节点的深度)返回。
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val; // 节点存储的整数值
* TreeNode *left; // 指向当前节点的左子节点的指针
* TreeNode *right; // 指向当前节点的右子节点的指针
* TreeNode() : val(0), left(nullptr), right(nullptr) {} // 默认构造函数
* TreeNode(int x) : val(x), left(nullptr), right(nullptr) {} // 带有节点值的构造函数
* TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {} // 完整的构造函数
* };
*/
class Solution {
public:
// 使用广度优先搜索计算二叉树的最大深度
int maxDepth(TreeNode* root) {
// 如果根节点为空,则深度为0
if(root == nullptr)
return 0;
// 创建一个队列用于存储每一层的节点
queue<TreeNode*> Q;
// 将根节点加入队列
Q.push(root);
// 初始化最大深度为0
int ans = 0;
// 当队列不为空时,继续循环
while(!Q.empty()){
// 获取当前层的节点数量
int s = Q.size();
// 遍历当前层的每个节点
while(s > 0){
// 取出队列中的一个节点
TreeNode* temp = Q.front();
Q.pop();
// 如果该节点的左子节点不为空,则将其加入队列
if(temp->left != nullptr)
Q.push(temp->left);
// 如果该节点的右子节点不为空,则将其加入队列
if(temp->right != nullptr)
Q.push(temp->right);
// 处理完一个节点,当前层的节点数量减1
s--;
}
// 当前层的所有节点都被处理完毕,深度加1
ans++;
}
// 返回最大深度
return ans;
}
};
在这个解决方案中,我们使用了广度优先搜索(BFS)的方法来计算二叉树的最大深度。我们维护一个队列,初始时只包含根节点。在每一轮中,我们将队列中的所有节点都取出,并将它们的子节点(如果有的话)加入队列。这样每向队列中添加一层节点,深度计数器 ans
就加1。当队列为空时,表示所有节点都已经被处理,此时 ans
就是最大深度。