【算法笔记】求二叉树节点个数——层序遍历法

最新推荐文章于 2025-04-30 16:36:16 发布

Cuteboom

最新推荐文章于 2025-04-30 16:36:16 发布

阅读量125

点赞数

文章标签：算法笔记 java

本文链接：https://blog.youkuaiyun.com/Cuteboom/article/details/131603154

版权

该代码实现了一个函数，通过队列进行广度优先搜索遍历二叉树，计算并返回所有节点的数量。程序首先初始化一个队列，然后从根节点开始，每次出队一个节点并递增节点计数，同时将非空的孩子节点入队，直到队列为空。最后，返回节点总数。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

int findallcodenumber(bitree root) {
    if (root == NULL)
        return 0;

    Queue q;
    initqueue(q); // 初始化队列
    enqueue(q, root); // 将根节点入队
    int codenumber = 0;

    while (!isempty(q)) {
        bitree current;
        dequeue(q, current);
        codenumber++;

        if (current->lchild != NULL)
            enqueue(q, current->lchild); // 将左孩子入队

        if (current->rchild != NULL)
            enqueue(q, current->rchild); // 将右孩子入队
    }

    return codenumber;
}

完整代码

#include <iostream>
#include <queue>

using namespace std;

// 二叉树节点结构
struct TreeNode {
    int data;               // 节点数据
    struct TreeNode* lchild;  // 左子节点指针
    struct TreeNode* rchild;  // 右子节点指针
};

typedef struct TreeNode* bitree;   // 定义二叉树指针类型
typedef queue<bitree> Queue;       // 定义队列类型

void initqueue(Queue& q) {
    while (!q.empty()) {
        q.pop();    // 清空队列
    }
}

void enqueue(Queue& q, bitree item) {
    q.push(item);   // 入队操作
}

void dequeue(Queue& q, bitree& item) {
    item = q.front();   // 获取队头节点
    q.pop();            // 出队操作
}

bool isempty(Queue& q) {
    return q.empty();   // 判断队列是否为空
}

int findallcodenumber(bitree root) {
    if (root == NULL)
        return 0;

    Queue q;
    initqueue(q); // 初始化队列
    enqueue(q, root); // 将根节点入队
    int codenumber = 0;

    while (!isempty(q)) {
        bitree current;
        dequeue(q, current);
        codenumber++;

        if (current->lchild != NULL)
            enqueue(q, current->lchild); // 将左孩子入队

        if (current->rchild != NULL)
            enqueue(q, current->rchild); // 将右孩子入队
    }

    return codenumber;
}

bitree createNode(int data) {
    bitree newNode = new TreeNode;
    newNode->data = data;
    newNode->lchild = NULL;
    newNode->rchild = NULL;
    return newNode;
}

bitree createBinaryTree() {
    bitree root =createNode(1) ;
    root->lchild = createNode(2);
    root->rchild = createNode(3);
    root->lchild->lchild = createNode(4);
    root->lchild->rchild = createNode(5);
    root->rchild->lchild = createNode(6);
    root->rchild->rchild = createNode(7);
    return root;
}

int main() {
    bitree root = createBinaryTree();
    int totalNodes = findallcodenumber(root);
    cout << "二叉树中节点的总数为：" << totalNodes << endl;

    return 0;
}