Update Lab7

2026-06-09 20:12:24 +00:00 · 2018-11-06 20:54:44 +07:00
parent a01379c222
commit 5b6f43c7ea
5 changed files with 375 additions and 0 deletions
@@ -0,0 +1,80 @@
+#include "binarysearchtree.h"
+
+void BinarySearchTree::insert(TreeNode *node)
+{
+    if (this->root == nullptr) {
+        this->root = node;
+        return;
+    }
+    TreeNode *temp = this->root;
+    while (true) {
+        if (temp->getData() > node->getData()) {
+            if (temp->getLeft() == nullptr) {
+                temp->setLeft(node);
+                return;
+            }
+            temp = temp->getLeft();
+        } else if (temp->getData() < node->getData()) {
+            if (temp->getRight() == nullptr) {
+                temp->setRight(node);
+                return;
+            }
+            temp = temp->getRight();
+        } else {
+            return;
+        }
+    }
+}
+
+void BinarySearchTree::remove(int removeData)
+{
+    if (this->root == nullptr)
+        return;
+    TreeNode *preTemp = nullptr;
+    TreeNode *temp = this->root;
+    while (temp != nullptr) {
+        if (temp->getData() > removeData) {
+            preTemp = temp;
+            temp = temp->getLeft();
+        } else if (temp->getData() < removeData) {
+            preTemp = temp;
+            temp = temp->getRight();
+        } else {
+            break;
+        }
+    }
+    if (temp != nullptr) {
+        if (temp->getLeft() != nullptr) {
+            TreeNode *preMaxL = nullptr;
+            TreeNode *maxL = temp->getLeft();
+            while (maxL->getRight() != nullptr) {
+                preMaxL = maxL;
+                maxL = maxL->getRight();
+            }
+            temp->setData(maxL->getData());
+            if (preMaxL != nullptr)
+                preMaxL->setRight(maxL->getLeft());
+            else
+                temp->setLeft(maxL->getLeft());
+            delete maxL;
+        } else {
+            if (preTemp != nullptr) {
+                if (preTemp->getLeft() == temp)
+                    preTemp->setLeft(temp->getRight());
+                else
+                    preTemp->setRight(temp->getRight());
+            } else {
+                this->root = temp->getRight();
+            }
+            delete temp;
+        }
+    }
+}
+
+void BinarySearchTree::print()
+{
+    if (this->root == nullptr)
+        return;
+    this->root->print();
+    cout << endl;
+}