Fix llfree removing nodes

2 files changed, 9 insertions, 6 deletions

diff --git a/src/ll.c b/src/ll.c
index fdf86af..8e2d078 100644
--- a/src/ll.c
+++ b/src/ll.c
@@ -70,15 +70,18 @@ void llinsert(Node** head, int value, int index) {
   node->next = n;
 }
 
-void llfree(Node* head) {
-  Node* n = NULL;
-  while (head != NULL) {
-    n = head;
-    head = head->next;
+void llfree(Node** head) {
+  while (*head != NULL) {
+    Node* n = *head;
+    *head = (*head)->next;
     free(n);
   }
+  *head = NULL;
 }
 
+// TODO: llfree recursive implementation
+// ...
+
 int llrmlast(Node** head) {
   Node* n = *head;
   Node* cur = NULL;
diff --git a/src/ll.h b/src/ll.h
index f777ace..b4c8cc9 100644
--- a/src/ll.h
+++ b/src/ll.h
@@ -41,7 +41,7 @@ void llinsert(Node** head, int value, int index);
 
 // Free nodes from memory
 // O(n)
-void llfree(Node* head);
+void llfree(Node** head);
 
 // Remove last node
 // O(n)