free node for item removed from queue

1 files changed, 3 insertions, 1 deletions

diff --git a/src/01/01a/priority_queue.c b/src/01/01a/priority_queue.c
index 35cfa92..adaf5ea 100644
--- a/src/01/01a/priority_queue.c
+++ b/src/01/01a/priority_queue.c
@@ -52,9 +52,11 @@ void add(PriorityQueue *queue, int priority, int data) {
 int delete_min(PriorityQueue *queue) {
   if (queue->head) {
     Node *tmp = queue->head;
+    int data = tmp->data;
     queue->head = tmp->next;
     queue->size--;
-    return tmp->data;
+    free(tmp);
+    return data;
   }
   return 0;
 }