小鑫的算法之路：leetcode0104 二叉树的最大深度

题目

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

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

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

解法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)。执行结果如下：