104.二叉树的最大深度（队列层序遍历+递归）

最新推荐文章于 2022-06-30 15:23:57 发布

asjodnobfy

最新推荐文章于 2022-06-30 15:23:57 发布

阅读量77

点赞数

分类专栏： C++ LeetCode 文章标签： c++ 深度优先开发语言

本文链接：https://blog.csdn.net/qq_41550190/article/details/122378956

版权

LeetCode 同时被 2 个专栏收录

61 篇文章 0 订阅

订阅专栏

C++

26 篇文章 0 订阅

订阅专栏

104.二叉树的最大深度

解题思路1

借助队列，遍历完一层，层数+1

代码

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
    int maxDepth(TreeNode* root) {
        queue<TreeNode*> que;
        int  dep = 0;

        if(root != NULL) que.push(root);
        while(!que.empty()){
            //每层元素个数
            int size = que.size();
            for(int i=0;i<size;i++){
                //弹出队头元素
                TreeNode* node = que.front();
                que.pop();
                //左右子树入队
                if(node->left) que.push(node->left);
                if(node->right) que.push(node->right);
            }
            //结束一层，层数+1
            dep += 1;
        }
        return dep;
    }
};

执行用时：8 ms, 在所有 C++ 提交中击败了66.65%的用户
内存消耗：18.4 MB, 在所有 C++ 提交中击败了43.96%的用户
通过测试用例：39 / 39

解题思路2

递归

代码

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
    //递归1：返回类型int，参数node
    int getdepth(TreeNode* node){
        if(node == NULL) return 0;  //递归2：终止条件

        //递归3：单层调用逻辑
        //左右子树自己调用自己
        int leftdepth = getdepth(node->left);   
        int rightdepth = getdepth(node->right);
        //确定最大深度
        int depth = 1 + max(leftdepth,rightdepth);
        return depth;
    }
    int maxDepth(TreeNode* root) {
        return getdepth(root);
    }
};

精简以后：

代码

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
    //递归精简版
    int maxDepth(TreeNode* root) {
        //为空判断
        if(root == NULL) return 0;
        //自己调用自己，取最大深度+1
        return 1 + max(maxDepth(root->left),maxDepth(root->right));
    }
};