104. Maximum Depth of Binary Tree

最新推荐文章于 2024-10-15 15:32:42 发布

原创最新推荐文章于 2024-10-15 15:32:42 发布 · 442 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#binarytree #递归 #DFS #BFS

leetcode 专栏收录该内容

1 篇文章

订阅专栏

本文介绍了三种不同的算法实现方式来求解二叉树的最大深度：递归、深度优先搜索（DFS）及广度优先搜索（BFS）。递归方法直接通过函数调用计算左右子树的深度；DFS使用栈来迭代遍历树节点并记录深度；BFS则利用队列进行层次遍历，同样记录节点深度。

递归

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

DFS

class Solution {
public:
    int maxDepth(TreeNode* root) {
        if(root==NULL)
            return 0;
        stack<TreeNode*> nodes; //save traversal nodes
        int result=0;   //save max depth of node
        nodes.push(root);
        root->val=1;
        while(nodes.size()>0)
        {
            TreeNode* top=nodes.top();
            nodes.pop();
            int curDepth=top->val;
            result=result>curDepth?result:curDepth;
            if(top->left!=NULL)
            {
                nodes.push(top->left);
                top->left->val=curDepth+1;
            }
            if(top->right!=NULL)
            {
                nodes.push(top->right);
                top->right->val=curDepth+1;
            }
        }
        return result;
    }
};

BFS

class Solution {
public:
    int maxDepth(TreeNode* root) {
        if(root==NULL)
            return 0;
        queue<TreeNode*> nodes;
        nodes.push(root);
        root->val=1;
        int result=0;
        while(nodes.size())
        {
            TreeNode *fro=nodes.front();
            int curdepth=fro->val;
            nodes.pop();
            result=result>curdepth?result:curdepth;
            if(fro->left)
            {
                nodes.push(fro->left);
                fro->left->val=curdepth+1;
            }
            if(fro->right)
            {
                nodes.push(fro->right);
                fro->right->val=curdepth+1;
            }
        }
        return result;
    }
};