C語言依據使用者輸入的資料大小動態開闢記憶體空間

前言

使用C語言程式設計時經常需要從各種途徑讀取資料並將其儲存在記憶體中,但C語言為了保持簡潔,所以只包含一些很基礎的資料型別,有時候並不能很好的儲存使用者輸入的資料。
例如:需要儲存使用者輸入的字串,通常,教科書上所教的方法是分配一個較大的陣列來進行儲存。顯然,這在實際生產環境中是相當不切實際的。
為了相對完美的解決這個問題,我們可以透過動態的開闢堆空間來實現對資料的儲存。

例程簡介

此例程每次接收一個字元,並利用realloc函式根據資料量大小來重新分配堆空間。

例程程式碼

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

int main(void) {
    char tmp = '\0';
    char *str = NULL;
    
    for (int i = 0; true; i++) {
        tmp = getchar();
        
        if ((tmp == '\r') || (tmp == '\n'))
            break; /*每次迴圈都按照當前資料量大小來重新分配堆空間大小*/
            
        str = (char *) realloc(str, sizeof(str) + sizeof(tmp));
        
        if (str == NULL)
            return -1;
        
        str[i] = tmp;
    }
    
    printf("%s\n", str);
    
    return 0;
}

執行結果

輸入:hello,it7e!!!,程式動態開闢記憶體儲存字串並輸出hello,it7e!!!,如下圖。
2018-02-10 17-53-23螢幕截圖.png