fix: memory leak in huffman.cpp (#2728)

TheAlgorithms · Sep 11, 2024 · d74f4d3 · d74f4d3
1 parent 15e3fed
commit d74f4d3
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/greedy_algorithms/huffman.cpp b/greedy_algorithms/huffman.cpp
@@ -23,6 +23,14 @@ struct MinHeapNode {
     }
 };
 
+void deleteAll(const MinHeapNode* const root) {
+    if (root) {
+        deleteAll(root->left);
+        deleteAll(root->right);
+        delete root;
+    }
+}
+
 // For comparison of
 // two heap nodes (needed in min heap)
 struct compare {
@@ -85,6 +93,7 @@ void HuffmanCodes(char data[], int freq[], int size) {
     // Print Huffman codes using
     // the Huffman tree built above
     printCodes(minHeap.top(), "");
+    deleteAll(minHeap.top());
 }
 
 // Driver program to test above functions