104. 二叉树的最大深度
思路:广度优先搜索
题目:给定一个二叉树,找出其最大深度。二叉树的深度为根节点到最远叶子节点的最长路径上的节点数。
说明: 叶子节点是指没有子节点的节点。
示例:
给定二叉树 [3,9,20,null,null,15,7],
3
/
9 20
/
15 7
返回它的最大深度 3 。
可以用「广度优先搜索」的方法来解决这道题目,广度优先搜索的队列里存放的是「当前层的所有节点」。每次拓展下一层的时候,不同于广度优先搜索的每次只从队列里拿出一个节点,我们需要将队列里的所有节点都拿出来进行拓展,这样能保证每次拓展完的时候队列里存放的是当前层的所有节点,即我们是一层一层地进行拓展,最后我们用一个变量 ans来维护拓展的次数,该二叉树的最大深度即为 ans
class Solution {
public:
int maxDepth(TreeNode* root) {
//若根节点为空,直接返回
if(root == NULL) return 0;
//定义一个队列,并将题给的根节点入队
queue<TreeNode *> Q;
Q.push(root);
int ans = 0;//ans记录遍历的层数
//当队列不为空时
while(!Q.empty())
{
int sz = Q.size();
//此时的sz为当前层结点个数,广度搜索需要将每一层的节点遍历完
while(sz > 0)
{
TreeNode * node = Q.front();
Q.pop();
//当前结点的左右子节点都存在时,入队
if(node->left) Q.push(node->left);
if(node->right) Q.push(node->right);
sz--;
}
ans++;
}
return ans;
}
};
时间复杂度:O(n),其中 n 为二叉树的节点个数。与方法一同样的分析,每个节点只会被访问一次。
空间复杂度:此方法空间的消耗取决于队列存储的元素数量,其在最坏情况下会达到 O(n)。