From 4859bc0eccdae2c1d09bab91a7e3f0e6f523e95b Mon Sep 17 00:00:00 2001
From: Oleksandr Hryshchenko <oleksandr.hryshchenko@student.tuke.sk>
Date: Fri, 5 Nov 2021 08:40:52 +0000
Subject: [PATCH] =?UTF-8?q?=D0=98=D0=B7=D0=BC=D0=B5=D0=BD=D0=B8=D0=BB(?=
 =?UTF-8?q?=D0=B0)=20=D0=BD=D0=B0=20'cv4/a=5Ftrain.c'?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 cv4/a_train.c | 29 ++++++++++++++++++++++-------
 1 file changed, 22 insertions(+), 7 deletions(-)

diff --git a/cv4/a_train.c b/cv4/a_train.c
index 3b66395..25ab7b9 100644
--- a/cv4/a_train.c
+++ b/cv4/a_train.c
@@ -7,7 +7,7 @@ struct car* add_car(struct car* first, const char* target){
     if(!strcmp(target, ""))
         return NULL;
 
-    if(!first){
+    if(!first || !strcmp(first->value, "")){
         first = (struct car*)calloc(1, sizeof(struct car));
         strcpy(first->value, target);
         first->next = NULL;
@@ -16,12 +16,12 @@ struct car* add_car(struct car* first, const char* target){
 
     struct car* temp = first;
 
-    while(temp->next != NULL)
+    while(temp != NULL)
         temp = temp->next;
 
-    temp->next = (struct car*)calloc(1, sizeof(struct car));
-    strcpy(temp->next->value, target);
-    temp->next->next = NULL;
+    temp = (struct car*)calloc(1, sizeof(struct car));
+    strcpy(temp->value, target);
+    temp->next = NULL;
     return first;
 }
 
@@ -59,12 +59,27 @@ struct car* clear_train(struct car* first, const char* target){
 
     while(temp != NULL){
         next = temp->next;
-        if(!strcmp(temp->value, target))
-        free(temp); 
+        if(!strcmp(temp->value, target)) {
+            if(temp == first) {
+                free(first);
+                first = NULL;
+            }
+            else
+                free(temp);
+        }
         temp = next;
         if(!first)
             first = temp;
     }
 
     return first;
+}
+
+int main(){
+    struct car* myCar = (struct car*)calloc(1, sizeof(struct car));
+    myCar = add_car(myCar, "Kosice");
+    print_train(myCar);
+    myCar = clear_train(myCar, "Kosice");
+    print_train(myCar);
+    return 0;
 }
\ No newline at end of file