基于类模板实现的二叉树基本操作,中序遍历实现数据有序输出

这篇博客介绍了如何使用C++实现二叉搜索树(BST)的基本操作,包括插入、删除节点以及中序遍历。同时,定义了一个`Student`类模板,实现了比较操作符,用于在BST中存储和比较学生对象。示例展示了如何创建包含学生分数的BST并进行操作。
摘要由CSDN通过智能技术生成

话不多说,直接上代码

#include <iostream>

using namespace std;

template<typename T>
struct Node{
    T val;
    Node * left, *right;
    Node(T v):val(v), left(nullptr), right(nullptr){}
};

template<typename T>
void printBst(Node<T> * root){
    if(root == nullptr) return;
    //先序遍历
    printBst(root->left);
    //中序遍历
    cout <<" "<< root->val << " ";
    printBst(root->right);
    //后序遍历
}

template<typename T>
Node<T>* insert(Node<T> * node, T val){
    if(node == nullptr) return new Node<T>(val); 
    if(node->val < val) node->right = insert(node->right, val);
    else if(node->val > val) node->left = insert(node->left, val);
    return node;
}

template<typename T>
Node<T>* minRightT(Node<T> * node){
        //此处严禁更改node的指向,会乱套
    Node<T> * current = node;
    while(current->left != nullptr)
        current = current->left;
    return current;
}

template<typename T>
Node<T>* deletNode(Node<T> * node, T val){
    if(node == nullptr) return node;
    if(node->val > val) node->left = deletNode(node->left, val);
    else if(node->val < val) node->right = deletNode(node->right, val);
    else{
        if(node->left == nullptr && node->right == nullptr) {delete node; return nullptr;}
        else if(node->left != nullptr && node->right != nullptr){
            Node<T> * minR = minRightT(node->right);
            node->val = minR->val;
            node->right = deletNode(node->right, minR->val);//这里的参数一定要用node->right保持树的根节点
        }
        else{
            Node<T> *temp = (node->left == nullptr ? node->right : node->left);
            delete node;
            return temp;
        }
    }
    return node;
}

class Student
{
    public:
    Student() {cout <<"默认Constructor"<< endl;}
    Student(int score) {this->score = score;}
    Student(const Student &a) {this->score = a.score;}
    int getScore(){return this->score;}
    void setScore(int score){this->score = score;}
    bool operator<(const Student &a){return this->score < a.score;}
    bool operator>(const Student &a){return this->score > a.score;}
    bool operator>=(const Student &a){return !(*this<a);}//由于优先级,注意要加括号
    void operator=(const Student &a){this->score = a.score;}
    bool operator==(const Student &a){return this->score == a.score;}
    friend ostream & operator<<(ostream & os, const Student &a);//注意要使用friend友元
    private:
    int score;
};

ostream & operator<<(ostream & os, const Student &a){
    os<<"score:"<<a.score; return os;
}

int main(){
    Node<int>* root = nullptr;
    root = insert(root, 50);
    insert(root, 30);
    insert(root, 20);
    insert(root, 40);
    insert(root, 70);
    insert(root, 60);
    insert(root, 80);

    printBst(root);
    cout << endl;
    root = deletNode(root, 80);
    printBst(root);
    root = deletNode(root, 50);
    cout << endl;
    printBst(root);
    cout<<endl;

    Node<Student>* rootStu = nullptr;
    rootStu = insert(rootStu, Student(50));
    insert(rootStu, Student(30));
    insert(rootStu, Student(20));
    insert(rootStu, Student(40));
    insert(rootStu, Student(70));
    insert(rootStu, Student(60));
    insert(rootStu, Student(80));
    printBst(rootStu);
    cout << endl;
}

新人一枚,错误和纰漏在所难免,还请各位圈中大佬指点一二。

  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值