LeetCode104—Maximum Depth of Binary Tree
求二叉树的深度。
原题
Given a binary tree, find its maximum depth.
The maximum depth is the number of nodes along the longest path from the root node down to the farthest leaf node.
两种解法
深度优先搜索
每次递归深度加1
代码一
class Solution {
private:
int dfs(TreeNode* root, int depth, int maxDepth)
{
if (root == NULL)
{
if (depth > maxDepth)
maxDepth = depth;
return maxDepth;
}
return max(dfs(root->left, depth + 1, maxDepth), dfs(root->right, depth + 1, maxDepth));
}
public:
int maxDepth(TreeNode* root) {
int depth = dfs(root, 0, INT_MIN);
return depth;
}
};
网上大神写了一种更简洁的代码
代码二
class Solution{
public:
int maxDepth(TreeNode* root) {
if (root == NULL)
return 0;
return max(maxDepth(root->left), maxDepth(root->right)) + 1;
}
};
层序遍历
BFS对二叉树进行层序遍历,记录遍历的层数即可
代码
class Solution{
public:
int maxDepth(TreeNode* root){
if (root == NULL)
return 0;
vector<TreeNode*>q;
int front = 0;
int rear = 1;
int level = 0;
q.push_back(root);
while (front < q.size())
{
rear = q.size();
while (front < rear)
{
if (q[front]->left != NULL)
{
q.push_back(q[front]->left);
}
if (q[front]->right != NULL)
{
q.push_back(q[front]->right);
}
++front;
}
++level;
}
return level;
}
};
本文介绍了解决LeetCode104题目——求二叉树最大深度的两种方法:深度优先搜索(DFS)及层序遍历(BFS)。深度优先搜索通过递归实现,而层序遍历采用队列实现。两种方法各有特点,可根据实际需求选择。
198

被折叠的 条评论
为什么被折叠?



