1.题目描述
输入一棵二叉树,求该树的深度。从根结点到叶结点依次经过的结点(含根、叶结点)形成树的一条路径,最长路径的长度为树的深度。
2.问题分析
方法一:
可以使用递归的方法,分别求左子树和右子树的深度,则树的深度是:更大者 + 1。这是一个递归的方法。
方法二:
使用层次遍历的方法,每访问一层,则深度+1,所有层的访问结束之后,就等到深度了。这里需要使用到一个队列,来帮助我们层次遍历,这不是一个递归的方法
3.源代码
方法一:
int TreeDepth(TreeNode* pRoot)
{
if(pRoot == NULL)
return 0;
int left = TreeDepth(pRoot->left);
int right = TreeDepth(pRoot->right);
return (left + 1 > right + 1) ? left + 1:right + 1;
}
方法二:
int TreeDepth(TreeNode* pRoot)
{
if(pRoot == NULL)
return 0;
queue<TreeNode*> nodes;
nodes.push(pRoot);
int deep = 0;
int curCount = 1;
int nextCount = 0;
while(!nodes.empty())
{
TreeNode* node = nodes.front();
nodes.pop();
--curCount;
if(node->left != NULL)
{
++nextCount;
nodes.push(node->left);
}
if(node->right != NULL)
{
++nextCount;
nodes.push(node->right);
}
if(curCount == 0)
{
curCount = nextCount;
nextCount = 0;
++deep;
}
}
return deep;
}