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