【东华大学oj】二叉树：建立存储结构（层次次序）

二叉树：建立存储结构（层次次序）

时间限制: 1s

类别: DS：树->中等

问题描述

目的：使用C++模板设计并逐步完善二叉树的抽象数据类型（ADT）。

内容：（1）请参照链表的ADT模板，设计二叉树并逐步完善的抽象数据类型。（由于该环境目前仅支持单文件的编译，故将所有内容都集中在一个源文件内。在实际的设计中，推荐将抽象类及对应的派生类分别放在单独的头文件中。参考教材、课件，以及网盘中的链表ADT原型文件，自行设计二叉树的ADT。）

注意：二叉树ADT的基本操作的算法设计很多要用到递归的程序设计方法。

 

（2）基本操作：二叉树的二叉链表存储形式的建立，完成后将其加入到二叉树的ADT基本操作集中。

输入数据为层次次序，要求设计一个算法，将二叉树转化为二叉链表的存储形式。

初始条件：definition给出二叉树T的定义（自然输入顺序序列。无孩子或指针为空的情形，算法通过特殊分隔符识别（输入）），至少有1个根结点。

输出：按definition构造二叉树的二叉链表。

注意：由于测试数据的显示需建立在二叉树的遍历基础上。因此，请在设计好二叉树的三种遍历算法之后（基本操作2），再进行测试。

参考函数原型：

//建立二叉树的存储结构 （用户函数）

template<class ElemType>

void CreateTree_Layer(BinaryTree<ElemType> &T, ElemType &str, ElemType &empty)；

输入说明

第一行：表示无孩子或指针为空的特殊分隔符

第二行：二叉树的层次次序序列（结点元素之间以空格分隔）

输出说明

第一行：二叉树先序遍历结果

第二行：二叉树中序遍历结果

第三行：二叉树后序遍历结果

 

#include <iostream>
#include <sstream>
#include <vector>
#include <queue>
#include <string>

using namespace std;

template<class ElemType>
struct BinaryTreeNode
{
    ElemType data;
    BinaryTreeNode<ElemType>* LChild;
    BinaryTreeNode<ElemType>* RChild;
    BinaryTreeNode<ElemType>* parent;

    BinaryTreeNode(ElemType item = ElemType(), BinaryTreeNode<ElemType>* L = nullptr, BinaryTreeNode<ElemType>* R = nullptr, BinaryTreeNode<ElemType>* P = nullptr)
        : data(item), LChild(L), RChild(R), parent(P) {}
};

template<class ElemType>
class BinaryTree
{
private:
    BinaryTreeNode<ElemType>* root;

    void destroy(BinaryTreeNode<ElemType>*& node)
    {
        if (node != nullptr)
        {
            destroy(node->LChild);
            destroy(node->RChild);
            delete node;
            node = nullptr;
        }
    }

    void preorder(BinaryTreeNode<ElemType>* node, vector<ElemType>& result)
    {
        if (node != nullptr)
        {
            result.push_back(node->data);
            preorder(node->LChild, result);
            preorder(node->RChild, result);
        }
    }

    void inorder(BinaryTreeNode<ElemType>* node, vector<ElemType>& result)
    {
        if (node != nullptr)
        {
            inorder(node->LChild, result);
            result.push_back(node->data);
            inorder(node->RChild, result);
        }
    }

    void postorder(BinaryTreeNode<ElemType>* node, vector<ElemType>& result)
    {
        if (node != nullptr)
        {
            postorder(node->LChild, result);
            postorder(node->RChild, result);
            result.push_back(node->data);
        }
    }

public:
    BinaryTree() : root(nullptr) {}
    ~BinaryTree()
    {
        destroy(root);
    }

    void createFromLevelOrder(vector<ElemType> elements, ElemType empty)
    {
        if (elements.empty() || elements[0] == empty)
            return;

        queue<BinaryTreeNode<ElemType>**> q;
        root = new BinaryTreeNode<ElemType>(elements[0]);
        q.push(&root);

        size_t i = 1;
        while (i < elements.size()) {
            BinaryTreeNode<ElemType>** parent = q.front();
            q.pop();

            if (elements[i] != empty) {
                (*parent)->LChild = new BinaryTreeNode<ElemType>(elements[i]);
                (*parent)->LChild->parent = *parent;
                q.push(&((*parent)->LChild));
            }
            ++i;

            if (i < elements.size() && elements[i] != empty) {
                (*parent)->RChild = new BinaryTreeNode<ElemType>(elements[i]);
                (*parent)->RChild->parent = *parent;
                q.push(&((*parent)->RChild));
            }
            ++i;
        }
    }

    BinaryTreeNode<ElemType>* convertToLinkedList() {
        if (root == nullptr) {
            return nullptr;
        }

        // 找到链表头节点
        BinaryTreeNode<ElemType>* head = root;
        while (head->LChild != nullptr) {
            head = head->LChild;
        }

        // 中序遍历构建链表
        BinaryTreeNode<ElemType>* current = head;
        while (current != nullptr) {
            if (current->RChild != nullptr) {
                BinaryTreeNode<ElemType>* next = current->RChild;
                while (next->LChild != nullptr) {
                    next = next->LChild;
                }
                next->LChild = current;
            }
            current = current->RChild;
        }

        return head;
    }

    void printPreorder()
    {
        vector<ElemType> result;
        preorder(root, result);
        printResult(result);
    }

    void printInorder()
    {
        vector<ElemType> result;
        inorder(root, result);
        printResult(result);
    }

    void printPostorder()
    {
        vector<ElemType> result;
        postorder(root, result);
        printResult(result);
    }

    void printResult(const vector<ElemType>& result)
    {
        if (!result.empty())
        {
            cout << result[0];
            for (size_t i = 1; i < result.size(); ++i)
            {
                cout << ',' << result[i];
            }
        }
        cout << endl;
    }
};

int main()
{
    string nullSymbol;
    string levelOrderInput;

    getline(cin, nullSymbol);
    getline(cin, levelOrderInput);

    stringstream ss(levelOrderInput);
    string item;
    vector<string> elements;

    while (ss >> item)
    {
        elements.push_back(item);
    }

    BinaryTree<string> tree;
    tree.createFromLevelOrder(elements, nullSymbol);

    tree.printPreorder();
    tree.printInorder();
    tree.printPostorder();

    return 0;
}