cv9

cv9/snake.c

@@ -21,7 +21,30 @@ struct snake* add_snake(struct snake* snake,int x,int y){
 }
 
 struct snake* remove_snake(struct snake* snake){
+    // Ak je zoznam hada prázdny, vrátiť NULL
+    if (snake == NULL) {
         return NULL;
+    }
+
+    // Ak máme iba jeden prvok, uvoľniť ho a vrátiť NULL
+    if (snake->next == NULL) {
+        free(snake);
+        return NULL;
+    }
+
+    // Prechádzanie zoznamu hada, kým nenájdeme predposledný prvok
+    struct snake* current = snake;
+    while (current->next->next != NULL) {
+        current = current->next;
+    }
+
+    // Uvoľnenie posledného prvku a nastavenie nového posledného prvku na NULL
+    free(current->next);
+    current->next = NULL;
+
+    // Vrátenie začiatku zoznamu hada
+    return snake;
+    //return NULL;
 }
 
 void free_snake(struct snake* sn){