非递归算法求二叉树高度

代码无bug

已于 2023-01-16 00:11:48 修改

阅读量366

点赞数

分类专栏：数据结构（树）文章标签：算法数据结构 c语言 c++ Powered by 金山文档

于 2023-01-15 17:24:18 首次发布

本文链接：https://blog.csdn.net/aaronyyds/article/details/128696018

版权

数据结构（树）专栏收录该内容

58 篇文章 0 订阅

订阅专栏

该文介绍了一种使用C++和队列容器实现的非递归算法，计算二叉树的高度。通过创建二叉链表结构，先序遍历输入数据，然后利用队列进行层次遍历来确定树的高度。

摘要由CSDN通过智能技术生成

假设二叉树采用二叉链表的存储结构，设计一个非递归算法求二叉树高度；

我这边用C++的队列容器解决。

#include<iostream>
#include<bits/stdc++.h>
using namespace std;
typedef struct BiNode {
    char data;
    struct BiNode* lchild, * rchild;
}BiNode, * BiTree;
void create(BiTree& t) {//二叉树先序
    char ch;
    cin >> ch;
    if (ch == '@')t = NULL;
    else {
        t = new BiNode;
        t->data = ch;
        create(t->lchild);
        create(t->rchild);
    }
}
int Btdepth(BiTree t) {//使用C++容器
    queue<BiTree>q;
    BiTree last = NULL;
    int num = 0;
    BiTree p = t;
    q.push(p);
    last = q.back();
    while (!q.empty()) {
        p = q.front();
       
        
        if (p->lchild)q.push(p->lchild);
        if (p->rchild)q.push(p->rchild);
        if (p == last) {
            num++;
            last = q.back();
        }
        q.pop();//这个出队要放在这边，不能放在p = q.front();的下一行，因为如果在p = q.front();下一行
        //出队后容器可能为空，p == last的判断的前提是容器不为空。
        //使用容器的 top()、back（）的使用,必须保证容器不为空


    }
    return num;
}
int main() {
    BiTree t;
    create(t);//测试数据：abd@@e@@c@@
    //abcde
    //edcba
    cout << Btdepth(t);
}

我重新又写了c语言的版本

#include<iostream>
using namespace std;
typedef struct BiNode {
    char data;
    struct BiNode* lchild, * rchild;
}BiNode, * BiTree;
typedef struct stack {
    BiTree data[1024];
    int top;
}stack;
typedef struct queue {
    BiTree data[1024];
    int front;
    int rear;
}queue;
void create(BiTree& t) {//二叉树先序
    char ch;
    cin >> ch;
    if (ch == '@')t = NULL;
    else {
        t = new BiNode;
        t->data = ch;
        create(t->lchild);
        create(t->rchild);
    }
}
int Btdepth(BiTree T) {
      BiTree q[1024];
    int front = 0, rear = 0;
    BiTree p = T;
    q[rear++] = p;
    int level = 0;
    int last = rear;
    while (front < rear) {
        p = q[front++];
        if (p->lchild)q[rear++] = p->lchild;
        if (p->rchild)q[rear++] = p->rchild;
        if (front == last) {
            level++;
            last = rear;
        }
    }
    return level;
}
int main() {
    BiTree t;
    create(t);//测试数据：abd@@e@@c@@
    //abcde
    //edcba
    cout << Btdepth(t);

}