Better memory management

Mazako · Mar 6, 2023 · 2e58239 · 2e58239
1 parent 7055748
commit 2e58239
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/double_linked_list/DoubleLinkedList.cpp b/double_linked_list/DoubleLinkedList.cpp
@@ -32,6 +32,7 @@ int DoubleLinkedList::pop() {
     if (size == 1) {
         int data = head->data;
         delete head;
+        head = nullptr;
         return data;
     }
     Node* nodePtr = head;
@@ -41,6 +42,7 @@ int DoubleLinkedList::pop() {
     int data = nodePtr->next->data;
     delete nodePtr->next;
     nodePtr->next = nullptr;
+    size--;
     return data;
 }
 
@@ -55,6 +57,9 @@ void DoubleLinkedList::printList() {
 
 void DoubleLinkedList::printReversed() {
     auto* nodePtr = head;
+    if (nodePtr == nullptr) {
+        return;
+    }
     while (nodePtr->next != nullptr) {
         nodePtr = nodePtr->next;
     }

diff --git a/main.cpp b/main.cpp
@@ -25,6 +25,7 @@ void simpleDoubleLinkedListTest() {
     list->pop();
     list->printList();
     list->printReversed();
+    delete list;
 }
 
 void simpleDynamicArrayTest() {