C++ 手动创建二叉树,并实现前序、中序、后序、层次遍历

二叉树的创建是个麻烦事,我的思路是:首先将一个普通的二叉树转化为满二叉树,其中的空节点用一些标识数据来代替,如此一来,就可以用数组索引来描述数据在二叉树的什么位置了。

比如,数组[2,4,3,1,5,-1,-1] 就可以表示一个三层的二叉树,具体长这样:

对于四种遍历方法,前序、中序、后序、层次遍历方法,给出了递归和迭代两种方式

具体代码如下:

#include <vector>
#include <iostream>
#include <algorithm>
#include <cmath>
#include <string>
#include <stack>
#include <queue>
using namespace std;
/*实现普通二叉树的创建,以及四种遍历方法*/
template <typename T>
class Node {
    public:
    T val = 0;
    Node<T>* lchild = nullptr;
    Node<T>* rchild = nullptr;
    Node(T& a): val(a){}
};

template <typename T>
class BinTree {
public:
    Node<T>* root = nullptr;
    BinTree(vector<T>& value, int n);
    void creatTree(Node<T>*& p, vector<T>& value, int n, const int len);
    void preOrder1(Node<T>* p);  //递归
    void preOrder2();   //迭代
    void inOrder1(Node<T>* p);  //递归
    void inOrder2();    //迭代
    void postOrder1(Node<T>* p);  //递归
    void postOrder2();   //迭代
    void levelOrder();
    void visit(Node<T>* p);
};

template <typename T>
BinTree<T>::BinTree(vector<T>& value, int n) {
    if ( value.size() != pow(2, n)-1 ) {//树的高度与元素个数不匹配
        cout << " elements not match to level " << endl;
        return;
    }
    creatTree(root, value, 0, pow(2, n)-1 );
}

template <typename T>
void BinTree<T>::creatTree(Node<T>*& p, vector<T>& value, int n, const int len) {
    if ( n < len && value[n] != -1 ) {
        p = new Node<T> (value[n]);
        creatTree(p->lchild, value, 2*n+1, len);    //创建左子树
        creatTree(p->rchild, value, 2*(n+1), len);  //创建you子树
    }
}

template <typename T>
void BinTree<T>::visit(Node<T>* p) {
    cout << p->val << " ";
}

template <typename T>
void BinTree<T>::preOrder1(Node<T>* p) {
    if (p) {
        visit(p);
        preOrder1(p->lchild);
        preOrder1(p->rchild);
    }
}

template <typename T>
void BinTree<T>::inOrder1(Node<T>* p) {
    if (p) {
        inOrder1(p->lchild);
        visit(p);
        inOrder1(p->rchild);
    }
}

template <typename T>
void BinTree<T>::postOrder1(Node<T>* p) {
    if (p) {
        postOrder1(p->lchild);
        postOrder1(p->rchild);
        visit(p);
    }
}

template <typename T>
void BinTree<T>::preOrder2() {
    stack<Node<T>*> st;
    Node<T>* p = this->root;
    st.push(p->rchild);
    visit(p);
    while ( !st.empty() ) {
        if ( p->lchild )
            p = p->lchild;
        else {
            p = st.top();
            st.pop();
        }
        visit(p);
        if ( p->rchild )
            st.push(p->rchild);
    }

}

template <typename T>
void BinTree<T>::inOrder2() {
    stack<Node<T>*> st;
    Node<T>* p = this->root;
    while ( p || !st.empty() ) {
        if (p) {
            st.push(p);
            p = p->lchild;
        }
        else {
            p = st.top();
            st.pop();
            visit(p);
            p = p->rchild;
        }
    }
}

template <typename T>
void BinTree<T>::postOrder2() {
    //太难了,cpu快想炸了,摆了
}

template <typename T>
void BinTree<T>::levelOrder() {
    queue<Node<T>*> Q;
    Node<T>* p = this->root;
    Q.push(p);
    while ( !Q.empty() ) {
        p = Q.front();
        Q.pop();
        visit(p);
        if ( p->lchild )
            Q.push(p->lchild);
        if ( p->rchild )
            Q.push(p->rchild);
    }
}

void sets(int* n) {
    n = new int (3);
}


int main() {
    vector<int> a = {2,4,3,1,5,-1,-1};
    a.shrink_to_fit();
    BinTree<int> *b = new BinTree<int> (a, 3);

    cout << b->root->val << " ";
    cout << b->root->lchild->val << " ";
    cout << b->root->rchild->val << " ";
    cout << b->root->lchild->lchild->val << " ";
    cout << b->root->lchild->rchild->val << endl;

    cout << "前序遍历(递归): ";
    b->preOrder1(b->root);
    cout << "\n中序遍历(递归): ";
    b->inOrder1(b->root);
    cout << "\n后序遍历(递归): ";
    b->postOrder1(b->root);

    cout << "\n前序遍历(迭代): ";
    b->preOrder2();
    cout << "\n中序遍历(迭代): ";
    b->inOrder2();

    cout << "\n层次遍历: ";
    b->levelOrder();
    cout << endl;
var code = "93e77dd0-c6ba-4c8a-bc02-31555a7fa65d"
}

以下是使用C++实现二叉树前序中序后序遍历的示例代码: ```cpp #include <iostream> using namespace std; // 定义二叉树结点 struct TreeNode { int val; TreeNode* left; TreeNode* right; TreeNode(int x) : val(x), left(nullptr), right(nullptr) {} }; // 构建二叉树 TreeNode* buildTree(int pre[], int in[], int preStart, int preEnd, int inStart, int inEnd) { if (preStart > preEnd || inStart > inEnd) { return nullptr; } int rootVal = pre[preStart]; int rootIndex = 0; for (int i = inStart; i <= inEnd; i++) { if (in[i] == rootVal) { rootIndex = i; break; } } int leftLen = rootIndex - inStart; TreeNode* root = new TreeNode(rootVal); root->left = buildTree(pre, in, preStart + 1, preStart + leftLen, inStart, rootIndex - 1); root->right = buildTree(pre, in, preStart + leftLen + 1, preEnd, rootIndex + 1, inEnd); return root; } // 前序遍历 void preOrder(TreeNode* root) { if (root == nullptr) { return; } cout << root->val << " "; preOrder(root->left); preOrder(root->right); } // 中序遍历 void inOrder(TreeNode* root) { if (root == nullptr) { return; } inOrder(root->left); cout << root->val << " "; inOrder(root->right); } // 后序遍历 void postOrder(TreeNode* root) { if (root == nullptr) { return; } postOrder(root->left); postOrder(root->right); cout << root->val << " "; } int main() { int pre[] = {1, 2, 4, 5, 3, 6, 7}; int in[] = {4, 2, 5, 1, 6, 3, 7}; TreeNode* root = buildTree(pre, in, 0, 6, 0, 6); cout << "前序遍历结果:"; preOrder(root); cout << endl; cout << "中序遍历结果:"; inOrder(root); cout << endl; cout << "后序遍历结果:"; postOrder(root); cout << endl; return 0; } ``` 在上面的代码中,我们首先定义了二叉树结点结构体`TreeNode`,然后实现了一个`buildTree`数,该数用于根据前序遍历数组和中序遍历数组构建二叉树。接着,我们分别实现前序遍历、中序遍历和后序遍历的数,最后在主数中调用这些数进行遍历。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值