关闭

二叉树中叶子节点的个数(非递归)

标签: 二叉树叶子节点个数非递归算法
378人阅读 评论(0) 收藏 举报
分类:

算法思想:在遍历二叉树时,判断当前访问的节点是不是叶子节点,然后对叶子节点求和即可。(前序、中序、后序、按层遍历均可)

以”前序”为例,代码中有详细注释,不再具体阐述,实现如下:

#include <iostream>
#define MAXSIZE 1000

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;

}


// 前序遍历
void PreOrderTraverse(BiTree T)
{
    if (T)
    {
        printf("%c\n", T->data);
        PreOrderTraverse(T->lchild);
        PreOrderTraverse(T->rchild);

    }
}


// 非递归,求二叉树中叶子节点的个数
int leaf_num(BiTree T)
{
    int top = -1; // 栈为空
    int count = 0;
    BiTree s[MAXSIZE]; // 申请一个栈空间

    while (T != NULL || top != -1) // 当前树的根节点不为空 或 栈不为空(说明需要继续遍历)
    {
        while (T != NULL) // 当前树的根节点不为空
        {
            if(T->lchild == NULL && T->rchild == NULL) // 若当前根节点的左右子树都为空,则是叶子节点
                count++;
            s[++top] = T; // 先 ++top,然后将当前的根节点入栈

            T = T->lchild; // 然后访问当前根节点的左子树

        }
        if (top != -1) // 若此时栈不为空(此时跳出了上面的while(T != NULL),说明了当前根节点的左子树为空)
        {
            T = s[top--]; // 先获取当前的根节点(即让当前的根节点出栈),然后 top--;
            T = T->rchild; // 然后访问当前根节点的右子树
        }
    }

    return count;
}


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

    BiTree T = createBiTree(); // 建立
    PreOrderTraverse(T); // 输出


    int num = leaf_num(T);
    printf("\n%d\n", num);

    return 0;
}
1
0

查看评论
* 以上用户言论只代表其个人观点,不代表CSDN网站的观点或立场
    个人资料
    • 访问:124538次
    • 积分:2619
    • 等级:
    • 排名:第13990名
    • 原创:135篇
    • 转载:13篇
    • 译文:0篇
    • 评论:11条
    文章分类
    最新评论