输入一棵二叉树,求该树的深度。从根结点到叶结点依次经过的结点(含根、叶结点)形成树的一条路径,最长路径的长度为树的深度。
最先想到的就是深搜,设置一个值维护当前访问的深度,遇到叶子结点就和全局最大值比较更新。
void TreeDepthCore(TreeNode* pNode, int currDep, int &maxDep)
{
if(pNode->right == NULL && pNode->left == NULL)
{
if(currDep > maxDep)
maxDep = currDep;
return;
}
if (pNode->left != NULL)
TreeDepthCore(pNode->left, currDep+1, maxDep);
if (pNode->right != NULL)
TreeDepthCore(pNode->right, currDep+1, maxDep);
}
int TreeDepthv2(TreeNode* pRoot)
{
if(pRoot == NULL)
return 0;
int maxDep = -10000;
int currDep = 0;
TreeDepthCore(pRoot, currDep, maxDep);
return maxDep+1;
}
还有一种就是递归求解,代码更少。也是剑指offer上给出的解法。
注意一个小地方,那就是当遍历到某一节点为null的时候,代表已经达到了叶子结点的下一层,此时要返回true,一方面空节点应该也是平衡的,另外只有返回true才能让上层与操作,否则每棵树都有遍历到null的时候,返回false,所有的结果都会变成false。
int TreeDepth(TreeNode* pRoot)
{
if(pRoot == NULL)
return 0;
int leftDepth = TreeDepth(pRoot->left);
int rightDepth = TreeDepth(pRoot->right);
return (leftDepth>rightDepth) ? (leftDepth+1):(rightDepth+1);
}