104. Maximum Depth of Binary Tree的C++解法

最新推荐文章于 2020-07-29 21:48:41 发布

原创最新推荐文章于 2020-07-29 21:48:41 发布 · 387 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode #C++

LeetCode 专栏收录该内容

158 篇文章

订阅专栏

本文介绍了使用递归和迭代两种方法来计算二叉树的最大深度。递归方法通过比较左右子树深度并加一实现，而迭代方法采用广度优先搜索进行遍历。这两种方法的时间复杂度相近。

用递归，当前树的深度就是左右子树中深度更深的那个+1。注意边界条件和判断子树为空的情况。

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    int maxDepth(TreeNode* root) {
        if (root==NULL) return 0;
        else return (1+max(maxDepth(root->left), maxDepth(root->right)));
    }
};

或者写得再简单一点，就是深度优先遍历。

int maxDepth(TreeNode *root)
{
    return root == NULL ? 0 : max(maxDepth(root -> left), maxDepth(root -> right)) + 1;
}

贴一个广度优先的遍历：

int maxDepth(TreeNode *root)
{
    if(root == NULL)
        return 0;
    
    int res = 0;
    queue<TreeNode *> q;
    q.push(root);
    while(!q.empty())
    {
        ++ res;
        for(int i = 0, n = q.size(); i < n; ++ i)
        {
            TreeNode *p = q.front();
            q.pop();
            
            if(p -> left != NULL)
                q.push(p -> left);
            if(p -> right != NULL)
                q.push(p -> right);
        }
    }
    
    return res;
}

时间复杂度好像区别不大。