diff --git a/cv9/snake.c b/cv9/snake.c index 92baaf5..8237939 100644 --- a/cv9/snake.c +++ b/cv9/snake.c @@ -3,10 +3,7 @@ struct snake* add_snake(struct snake* snake,int x,int y){ struct snake* hlavicka = calloc(1,sizeof(struct snake)); - if (hlavicka == NULL) { - return NULL; - } - + hlavicka->x = x; hlavicka->y = y; @@ -16,7 +13,15 @@ struct snake* add_snake(struct snake* snake,int x,int y){ } struct snake* remove_snake(struct snake* snake){ - return NULL; + + struct snake* nova_hlavicka = snake->next; + free(snake); + + if (nova_hlavicka != NULL) { + nova_hlavicka->next = NULL; + } + + return nova_hlavicka; } void free_snake(struct snake* sn){