剑指 Offer 55 - I. 二叉树的深度
输入一棵二叉树的根节点,求该树的深度。从根节点到叶节点依次经过的节点(含根、叶节点)形成树的一条路径,最长路径的长度为树的深度。
和题解写的一比,我写的像坨shit
//! 求二叉树的高度,递归版本
//! 递归的精髓就在于,把大问题分解为解答过程相同的小问题
//! 既然我们要求根节点的高度,根节点的高度又等于其子节点的最大高度加一
//! 那么问题就转化为了求每个节点及其子节点的高度,而最小问题就是叶节点的子节点为空,高度为0
int BinarySearchTreesZH::height(Node *node)
{
int nodeHeight = 0;
int leftSonHeight = 0; //代表左子节点高度
int rightSonHeight = 0; //代表右子节点高度
if (node == nullptr)
{ //如果到了空结点,则其高度为0
nodeHeight = 0;
return nodeHeight;
}
leftSonHeight = height(node->left);
rightSonHeight = height(node->right);
if (leftSonHeight >= rightSonHeight) //当前结点的高度等于其左右子节点最大高度+1
{
nodeHeight = leftSonHeight + 1;
}
else
{
nodeHeight = rightSonHeight + 1;
}
return nodeHeight;
}
//! 求二叉树的高度,迭代版本
//! 所谓二叉树的高度就是二叉树有多少层,所以在层序遍历里加一个层数计数就可以了
//! 但这个层数计数反倒是难想,其实在层序遍历的队列中,上一层全部出队后,下一层有多少元素,就是队列的size()
int BinarySearchTreesZH::heightNoRecursion(Node *node)
{
queue<Node *> list; //创建结构体指针队列,数据类型Node*,但是层序遍历不用指针也可以
if (node == nullptr)
{
return 0;
}
else
{
list.push(node); //把根节点push进去
}
int height = 0;
int nextCengNum = 1; //! 第一层元素数为1
int j = 0; //! 循环计数器,每一层循环nextCengNum次就会清零,并高度+1
while (list.size() != 0) //只要队列不为空就一直出队进队
{
//这里就是把队首的元素的左右子节点进队,然后弹出队首元素,就可以把二叉树中所有节点一层一层的按次序进队和出队
//还是画图比较好理解
if (list.front()->left != nullptr)
{
list.push(list.front()->left);
}
if (list.front()->right != nullptr)
{
list.push(list.front()->right);
}
list.pop();
j++; //这里开始就是计算遍历的层数的,遍历次数每到达下一层元素数时就代表一层
if (j == nextCengNum)
{
height++;
nextCengNum = list.size();
j = 0;
}
}
return height;
}