不论是“求二叉树第k层的节点数”还是“求二叉树第k层的叶子节点数”,算法思想都是二叉树的层序遍历,在前面几篇博文中已经多次讲述,此处就不再赘述了。
求二叉树第k层的叶子节点数,同样有递归方法和非递归方法,本文仅讲述递归方法,非递归方法可以参考前面几篇博文来实现。
说明:根节点位于0层,每层中的第一个节点的下标为0。
具体实现如下:
#include <iostream>
using namespace std;
typedef struct node
{
char data;
struct node *lchild;
struct node *rchild;
}BiNode, *BiTree;
// 先序建立二叉树 (输入时,按先序次序输入二叉树中结点的值,以 # 字符表示空树)
BiTree createBiTree()
{
BiTree T;
char c;
scanf("%c", &c);
if (c == '#')
T = NULL;
else
{
T = new BiNode; // 或 T = (BiTree)malloc(sizeof(BiNode));
T->data = c;
T->lchild = createBiTree();
T->rchild = createBiTree();
}
return T;
}
// 递归
// 参数level代表几层(不是第几层)(如2层,对应的是第3层)
int num = ; // level层的总节点数;
int leafNum = ; // level层的叶子节点数;
void PrintNodeAtLevel(BiTree T , int level)
{
if(T == NULL || level < ) return;
if(level == )
{
if (T->lchild == NULL && T->rchild == NULL)
{
leafNum++;
}
num++;
}
PrintNodeAtLevel(T->lchild , level - );
PrintNodeAtLevel(T->rchild , level - );
}
int main(int argc, const char * argv[]) {
BiTree T = createBiTree(); // 建立
PrintNodeAtLevel(T, ); // 2层叶子节点数(2层,是指第3层)
printf("该层的节点数为: %d, 该层叶子节点数为: %d\n", num, leafNum);
return ;
}