Как прочитать несколько строк из стандартного ввода и сохранить его в связанном списке - PullRequest
4 голосов
/ 04 июня 2019

Я пытаюсь написать программу, которая получает строки от пользователя (из STDIN) и сохраняет их в связанном списке.

сейчас я получаю только одну строку и прекращаю работу программы. Как я могу изменить код, чтобы получать строки из стандартного ввода?

также, если кто-то может сказать мне, выделяю ли я и освобождаю ли память, как это должно быть, это будет очень полезно.

Спасибо.

#include <stdio.h>
#include <stdlib.h>

int BUFF_SIZE = 128;

struct Node {
    char* data;
    struct Node* next;
};

struct Node* head = NULL;
struct Node* tail = NULL;

void free_list(struct Node* head)
{
    if (head != NULL)
    {
        free_list(head->next);
        free(head);
    }
}

int main()
{
    int curr_size = 0;

    char* pStr = malloc(BUFF_SIZE);
    curr_size = BUFF_SIZE;

    printf("%s", "please print multiple lines\n");
    if (pStr != NULL)
    {
        char c;
        int i = 0;

        while ((c = getchar()) != '\n' && c != EOF)
        {
            pStr[i++] = c;

            if (i == curr_size)
            {
                curr_size = i + BUFF_SIZE;
                pStr = realloc(pStr, curr_size);
                if (pStr == NULL) return;
            }
        }
        pStr[i] = '\0';

        struct Node* new_node = malloc(sizeof(struct Node*));
        char* new_data = malloc(sizeof(pStr));
        new_data = pStr;
        new_node->data = new_data;

        if (head == NULL)
        {
            head = new_node;
            tail = new_node;
        }

        else
        {
            tail->next = new_node;
        }
    }

    free_list(head);
}

1 Ответ

6 голосов
/ 04 июня 2019

Мало проблем:

  1. На данный момент вы прекращаете чтение после получения \n.

    if (pStr == NULL) return; //error
    
    int c;
    int i = 0;
    
    while ((c = getchar()) != EOF)
    {
       /*New word, insert into linked list*/
       if (c == '\n'){
           pStr[i] = '\0';
    
           struct Node* new_node = malloc(sizeof(*new_node));
           char* new_data = malloc(i+1);
           strcpy(new_data, pStr);
           new_node->data = new_data;
    
           if (head == NULL)
           {
                head = new_node;
                tail = new_node;
           }
           else
           {
                tail->next = new_node;
                tail = new_node;
           }
           i = 0; //Reset the index
       }
       else {
    
           pStr[i++] = c;
           if (i == curr_size)
           {
               curr_size = i + BUFF_SIZE;
               pStr = realloc(pStr, curr_size);
               if (pStr == NULL) return;
           }
       }
    }
    
  2. Утечки памятии узел data всегда будет указывать на последнее содержимое pStr.

    char* new_data = malloc(sizeof(pStr)); 
    new_data = pStr;   //Memory leak here
    new_node->data = new_data;
    

    , изменить его на

    char* new_data = malloc(i+1);
    strcpy(new_data, pStr);
    new_node->data = new_data;
    

    sizeof (pStr) - размер указателя недлина строки.

  3. Вам необходимо обновить tail после добавления каждого узла в список.

     else
     {
         tail->next = new_node;
     }
    

    до

     else
    {
        tail->next = new_node;
        tail = new_node;
     }
    
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...