如何预先确定malloc的大小以连接字符串数组？

Question

实现具有以下原型的append函数。 该函数返回一个字符串，该字符串表示存在于字符串数组中的所有字符串的串联。 对于此问题，您可以假定参数数组的结尾用NULL标记。 您需要为结果字符串分配内存。 您不能修改array参数。

 char* append(char *data[]); 

我不明白如何确定大小malloc的指针。

Answer 1

两次遍历字符串的方法似乎很好。

第一遍计算长度之和：

size_t len = 0;
for (char** pstr = data; *pstr; pstr++)
    len += strlen(*pstr);

第二遍将所有字符串连接起来：

char *str = malloc(len + 1);
str[0] = '\0';
for (char** pstr = data; *pstr; pstr++)
    strcat(str, *pstr);
return str;

您可以通过存储最后一个串联的终点来优化串联部分：

char *str = malloc(len + 1);
str[0] = '\0';

char *p = str;
for (char** pstr = data; *pstr; pstr++) {
    strcat(p, *pstr);
    p += strlen(*pstr);
}

Answer 2

首先，要了解一个字符串的大小，可以使用库string.h strlen 。 如果要计算所有大小的总和，则可以使用循环并加总所有strlen ，然后为终端NUL字符加1 ，如下所示：

char* append(char *data[]) {
    char **cur, *res;
    size_t len = 0;

    for (cur = data; *cur != NULL; *cur++)
        len += strlen(*cur);

    res = malloc(len + 1);

    // Now you can concatenate the strings...
}

哦，别忘了检查malloc返回的指针是否有效（即不是NULL ）。