二叉树第k层的叶子节点数

不论是“求二叉树第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 = 0; // level层的总节点数；
int leafNum = 0; // level层的叶子节点数；
void PrintNodeAtLevel(BiTree T , int level)
{
    if(T == NULL || level < 0) return;

    if(level == 0)
    {

        if (T->lchild == NULL && T->rchild == NULL)
        {
            leafNum++;
        }

        num++;

    }
    PrintNodeAtLevel(T->lchild , level - 1);
    PrintNodeAtLevel(T->rchild , level - 1);
}


int main(int argc, const char * argv[]) {

    BiTree T = createBiTree(); // 建立

    PrintNodeAtLevel(T, 2); // 2层叶子节点数（2层，是指第3层)

    printf("该层的节点数为: %d,  该层叶子节点数为: %d\n", num, leafNum);

    return 0;
}