add bst destructor

souravkr · Apr 19, 2020 · a758dad · a758dad
1 parent 65405d7
commit a758dad
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 0 deletions.
diff --git a/BinarySearchTree/BST/BST.cpp b/BinarySearchTree/BST/BST.cpp
@@ -1,4 +1,21 @@
 #include "BST.hpp"
+#include <iostream>
+
+/**
+ * Delete tree nodes recursively
+*/
+void BST::DeleteTreeRecursive(Node* node)
+{
+    if (node == nullptr)
+        return;
+
+    if (node->left != nullptr)
+        DeleteTreeRecursive(node->left);
+    if (node->right != nullptr)
+        DeleteTreeRecursive(node->right);
+
+    delete node;
+}
 
 /**
  * inserts new integer data into the tree at the position, so that

diff --git a/BinarySearchTree/BST/BST.hpp b/BinarySearchTree/BST/BST.hpp
@@ -40,9 +40,11 @@ class BST {
     Node* insert(Node* root, int data);
     bool search(Node* root, int data);
     int height(Node* root);
+    void DeleteTreeRecursive(Node* node);
 
 public:
     BST() { root = nullptr; }
+    ~BST() { DeleteTreeRecursive(root); }
     Node* getRoot() const { return root; }
     void insert(int data);
     bool search(int data);