LeetCode——104. 二叉树的最大深度

一、题目

给定一个二叉树，找出其最大深度。

二叉树的深度为根节点到最远叶子节点的最长路径上的节点数。

说明: 叶子节点是指没有子节点的节点。

来源：力扣（LeetCode）
链接：https://leetcode.cn/problems/maximum-depth-of-binary-tree/description/

二、C++解法

我的思路及代码

采用BFS，一层一层的对二叉树进行遍历，当整个树都遍历完成的时候也即是当队列中不存在元素的时候，返回即可。

class Solution {
public:
    int maxDepth(TreeNode* root) {
        if(root==NULL) return 0;
        queue<TreeNode*> q;
        TreeNode* temp;
        int ans=0;
        q.push(root);

        while(q.size()){
            int size = q.size();
            for(int i=0;i<size;i++){
                temp = q.front();
                q.pop();
                if(temp->left!=NULL)
                    q.push(temp->left);
                if(temp->right!=NULL)
                    q.push(temp->right);
            }
            ans++;
        }
        return ans;
    }
};

• 时间复杂度：O(n)，其中 n 为二叉树的节点个数。与方法一同样的分析，每个节点只会被访问一次
• 空间复杂度：此方法空间的消耗取决于队列存储的元素数量，其在最坏情况下会达到 O(n)

官方参考代码

由于我写了BFS的做法，所以这里放上DFS的做法

class Solution {
public:
    int maxDepth(TreeNode* root) {
        if (root == nullptr) return 0;
        return max(maxDepth(root->left), maxDepth(root->right)) + 1;
    }
};

• 时间复杂度：O(n)，其中 n 为二叉树节点的个数。每个节点在递归中只被遍历一次
• 空间复杂度：O(height)，其中 height 表示二叉树的高度。递归函数需要栈空间，而栈空间取决于递归的深度，因此空间复杂度等价于二叉树的高度