问题描述
输入一棵二叉树,求该树的深度。从根结点到叶结点依次经过的结点(含根、叶结点)形成树的一条路径,最长路径的长度为树的深度。
解题分析
根据深度的描述,我们可以直接借鉴深度优先遍历的方法,分别计算左子树和右子树的深度,然后取其较大的一个再加上根的长度(1)即可,具体代码可以参考下文的DFS。
还有一个比较好理解的方法就是基于层序遍历实现的,因为二叉树的深度等于其层数,所以借助队列进行层序遍历,然后记下层数即可,具体代码可以参考下文的BFS。
主要代码 c++
DFS 递归深度优先
class Solution {
public:
int TreeDepth(TreeNode* pRoot)
{
if(pRoot == nullptr) return 0;
int left = TreeDepth(pRoot->left);
int right = TreeDepth(pRoot->right);
return max(1+left, 1+right);
}
};
BFS 循环广度优先,基于层序遍历
class Solution {
public:
int TreeDepth(TreeNode* pRoot)
{
if(pRoot == nullptr) return 0;
queue<TreeNode*>res;
res.push(pRoot);
int depth = 0;
while(!res.empty())
{
int size = res.size();
for(int i = 0; i < size; ++i)
{
TreeNode* tmp = res.front(); res.pop();
if(tmp->left) res.push(tmp->left);
if(tmp->right) res.push(tmp->right);
}
depth++;
}
return depth;
}
};
相似题目:
解答 leetcode 104. Maximum Depth of Binary Tree (求二叉树的最大深度,DFS,BFS)
解答 leetcode 111. Minimum Depth of Binary Tree(求二叉树的最小深度,DFS,BFS)
解答 leetcode 559 Maximum Depth of N-ary Tree(N叉树的最大深度