如何使用C++实现二叉搜索树

YT8984

于 2023-06-12 20:26:58 发布

阅读量822

点赞数 1

文章标签： c++ 数据结构开发语言

YT8984：想要就拿去！不用谢，发文章记得点转载哦！有任何问题私信我。

本文链接：https://blog.csdn.net/yt21198/article/details/131176287

版权

本文介绍了如何使用C++实现二叉搜索树，包括节点定义、插入、查找、删除和遍历等操作。二叉搜索树是一种基于二分查找的数据结构，具有快速查找能力。文章还讨论了其性质、可能的退化问题及平衡二叉树（如AVL树和红黑树）的重要性。

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

二叉搜索树是一种基于二分查找的数据结构，它具有快速查找、插入和删除元素的能力。在C++中，可以用类来实现二叉搜索树，其节点包含左右子节点指针和数据值。下面介绍如何使用C++实现二叉搜索树。

定义节点

定义二叉搜索树节点的类。每个节点需要保存一个值、左右子节点指针：

class Node {
public:
    int value;
    Node* left;
    Node* right;

    Node(int v) {
        value = v;
        left = nullptr;
        right = nullptr;
    }
};

实现二叉搜索树类

定义二叉搜索树的类。它需要包含一个根节点指针和一些公共函数，如插入、查找、删除和遍历。

class BinarySearchTree {
public:
    Node* root;

    BinarySearchTree() {
        root = nullptr;
    }

    Node* insert(int value) {
        if (root == nullptr) {
            root = new Node(value);
            return root;
        } else {
            return insert(root, value);
        }
    }

    Node* insert(Node* node, int value) {
        if (node == nullptr) {
            node = new Node(value);
        } else if (value < node->value) {
            node->left = insert(node->left, value);
        } else if (value > node->value) {
            node->right = insert(node->right, value);
        }
        return node;
    }

    Node* search(int value) {
        return search(root, value);
    }

    Node* search(Node* node, int value) {
        if (node == nullptr || node->value == value) {
            return node;
        } else if (value < node->value) {
            return search(node->left, value);
        } else {
            return search(node->right, value);
        }
    }

    void remove(int value) {
        root = remove(root, value);
    }

    Node* remove(Node* node, int value) {
        if (node == nullptr) {
            return node;
        } else if (value < node->value) {
            node->left = remove(node->left, value);
        } else if (value > node->value) {
            node->right = remove(node->right, value);
        } else {
            if (node->left == nullptr) {
                Node* temp = node->right;
                delete node;
                return temp;
            } else if (node->right == nullptr) {
                Node* temp = node->left;
                delete node;
                return temp;
            }
            Node* successor = getMinNode(node->right);
            node->value = successor->value;
            node->right = remove(node->right, successor->value);
        }
        return node;
    }

    Node* getMinNode(Node* node) {
        while (node->left != nullptr) {
            node = node->left;
        }
        return node;
    }

    void inorderTraversal() {
        inorderTraversal(root);
    }

    void inorderTraversal(Node* node) {
        if (node != nullptr) {
            inorderTraversal(node->left);
            std::cout << node->value << " ";
            inorderTraversal(node->right);
        }
    }
};

测试二叉搜索树

使用一些测试用例测试二叉搜索树：

int main() {
    BinarySearchTree bst;

    // Insert nodes
    bst.insert(5);
    bst.insert(3);
    bst.insert(7);
    bst.insert(1);
    bst.insert(9);

    // Search nodes
    Node* node = nullptr;
    node = bst.search(3);
    if (node) std::cout << "Found: " << node->value << std::endl;

    // Remove nodes
    bst.remove(3);

    // Traverse nodes
    std::cout << "Inorder traversal: ";
    bst.inorderTraversal();
    std::cout << std::endl;

    return 0;
}

输出：