LCR 175. 计算二叉树的深度

LCR 175. 计算二叉树的深度

LCR 175

dfs解法

/**
 * 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 {
// 思路 :
// 一个树的最大深度=1+max(左子树的最大深度,右子树的最大深度)   
// 这个1是它root这一层
int dfs(TreeNode*root){
    if(root==nullptr)return 0;//如果递归到了叶子节点的子树了,返回0
    return max(dfs(root->left),dfs(root->right) )+1;
}
public:
    int calculateDepth(TreeNode* root) {
        if(root==nullptr)return 0;///如果是空树,返回0
        return max(dfs(root->left),dfs(root->right) )+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 calculateDepth(TreeNode* root) {
        if(root==nullptr)return 0;
        return max(calculateDepth(root->left),calculateDepth(root->right) )+1;
    }
};

bfs解法

/**
 * 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 calculateDepth(TreeNode* root) {
        if(root==nullptr)return 0;//如果是空树,返回0
        int level=0;//当前层
        vector<TreeNode*>que;//队列
        que.push_back(root);
        while(!que.empty() ){//当前曾不是空层
            vector<TreeNode*>tmp;//存储当前层的下一层节点
            level++;
            for(TreeNode*it:que){//遍历当前层的下一层节点
                if(it->left!=nullptr)tmp.push_back(it->left);
                if(it->right!=nullptr)tmp.push_back(it->right);
            }
            que=tmp;//队列来到新的一层    
        }
        return level;//返回层数
    }
};