Aktualizovat du4/snake.c

du4/snake.c

@@ -18,29 +18,33 @@ struct snake* add_snake(struct snake* snake,int x,int y)
 
 struct snake* remove_snake(struct snake* snake){
 
+    //pomocne lok. premeny
     struct snake* currentPtr = snake;
-    struct snake* prevPtr;
+    struct snake* prevPtr = NULL;
     
     if (snake == NULL)
     {
         return NULL;
     }
-    while (1)
+
+    //kontrola, ze ci hned dalsi prvok je NULL rovno hned zo zaciatku..
+    if (snake->next == NULL)
     {
-        //pomocny pointer -> ulozenie predoslej casti/node-u
-        prevPtr = currentPtr;
-        //skontroluje, ze ci dalsia cast je PRAZDNA.
-        if (currentPtr->next == NULL)
-        {
-            //uprava listu -> odlinkje tuto cast (tak aby prevPtr->next NEFERENCOVAL na uz uvolnenu cast)
-            prevPtr->next = NULL;
-            //uvolni pamat pre akt. cast (uvolni pamat)
-            free(currentPtr);
-            break;
+        free(snake);
+        return NULL;
     }
-        //iteruj ku dalsiej casti (node)
+
+    //iteracia cez node-y
+    while (currentPtr->next != NULL)
+    {
+        prevPtr = currentPtr;
         currentPtr = currentPtr->next;
     }
+
+    //'currentPtr'= posledny platny node
+    prevPtr->next = NULL;
+    free(currentPtr);
+
     return snake;
 }