clear

cv5/a_train.c

@@ -51,18 +51,22 @@ struct car* clear_train(struct car* first, const char* target) {
     if(first == NULL){
         return 0;
     } 
-
+    
+    if(first->next == NULL){
+      int result = strcmp(first->value, target);
+        if(result == 0){
+           free(first);
+           return 0;          
+        }
+        else{
+            return first;
+        }
+    }
+     
     struct car* prev = first;
     while(prev->next != NULL){
         int x = strcmp(first->value, target);
         if(x == 0){
-            if(first->next == NULL){
-                free(first);
-                return NULL;
-            }
-            else{
-                return first;
-            }  
             prev = first->next;
             free(first);
             return prev;