题目描述
给出一个二叉树,要求算出第k层共有多少个节点并且将结果返回来
图示
当k为3时,返回3
额外要求:
1、给定树的节点数范围是 [1, 100]。
思路分析
要以分治子问题的思路分析,若需要第k层的节点个数,那就让左子树和右子树去统计个数,并且将结果返回来
子问题:转换成左子树的第k-1层和右子树的第k-1层
结束条件:k == 1且节点不为空,返回1
或者节点为空,返回0
除了到了那一层但是每节点的情况要考虑,还要考虑k不对劲的情况
样例代码
// 二叉树第k层结点个数
int BTreeLevelKSize(BTNode* root, int k)
{
//除了到了那一层但是每节点的情况要考虑,还要考虑k不对劲的情况
assert(k >= 1);
if (root == NULL)
{
return 0;
}
if (k == 1)
{
return 1;
}
else
{
return BTreeLevelKSize(root->left, k - 1) +
BTreeLevelKSize(root->right, k - 1);
}
}
总结
以上就是今天要讲的内容,本文简单介绍了(二叉树)计算第k层有多少个节点的解题思路,在力扣以及牛客网中还会有很多类似的题目,今后将会持续更新,敬请关注!