LCR 175. 计算二叉树的深度
LCR 175
dfs解法
cpp
复制代码
/**
* 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;
}
};
优化一下
cpp
复制代码
/**
* 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解法
cpp
复制代码
/**
* 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;//返回层数
}
};