DS树--二叉树高度

最新推荐文章于 2024-07-12 18:56:03 发布

鷸鰥

最新推荐文章于 2024-07-12 18:56:03 发布

阅读量390

点赞数 7

分类专栏：数据结构文章标签：算法数据结构 c++ 开发语言

本文链接：https://blog.csdn.net/m0_74919335/article/details/135743208

版权

数据结构专栏收录该内容

53 篇文章 7 订阅

订阅专栏

本文介绍了一个C++程序，使用先序遍历的方法来计算给定二叉树的高度。通过递归函数BTree_create_tree构建二叉树，BTree_high函数则计算并返回树的高度。

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

DS树–二叉树高度

题目描述

给出一棵二叉树，求它的高度。

注意，二叉树的层数是从1开始

输入

第一行输入一个整数t，表示有t个二叉树

第二行起输入每个二叉树的先序遍历结果，空树用字符‘0’表示，连续输入t行

输出

每行输出一个二叉树的高度

输入样例：

1
AB0C00D00

输出样例：

参考代码：

#include <iostream>
#include <stack>
#include <vector>
#include <queue>
#include <list>

using namespace std;

struct BiTreeNode {
    char data;
    BiTreeNode *leftChild, *rightChild;
};

string sTree;
int pos;

BiTreeNode *BTree_create_tree() {
    if (sTree[pos] == '0') {
        pos++;
        return nullptr;
    } else {
        auto *node = new BiTreeNode;
        node->data = sTree[pos];
        pos++;
        node->leftChild = BTree_create_tree();
        node->rightChild = BTree_create_tree();
        return node;
    }
}

BiTreeNode *Build() {
    cin >> sTree;
    pos = 0;
    return BTree_create_tree();
}

int BTree_high(BiTreeNode *root) {
    int sum = 0;
    if (root == nullptr)
        return 0;
    else {
        queue<BiTreeNode *> q;
        q.push(root);
        while (!q.empty()) {
            sum++;
            queue<BiTreeNode *> r;
            while (!q.empty()) {
                r.push(q.front());
                q.pop();
            }
            while (!r.empty()) {
                BiTreeNode *node = r.front();
                r.pop();
                if (node->leftChild != nullptr)
                    q.push(node->leftChild);
                if (node->rightChild != nullptr)
                    q.push(node->rightChild);
            }
        }
    }
    return sum;
}

int main() {
    int t;
    cin >> t;
    while (t--) {
        BiTreeNode *root;
        root = Build();
        cout << BTree_high(root) << endl;
    }
    return 0;
}