使用动态内存分配设置 char 数组的确切大小

Question

我必须从标准输入中读取一些字符（我不知道其中有多少，但不超过 MAX）并且我想将它们存储在一个数组中。这段代码 sn-p 正确吗？

char *c1 = (char*) malloc(MAX * sizeof(char)); //may be too much
fgets(c, MAX, stdin);
int size = strlen(c1);
char *c2 = (char*) realloc(c1, size * sizeof(char));
free(c1);

当您不知道要存储多少元素时，是否有更优雅的方法来确定为数组分配的正确内存量？

Answer 1

在fgets() 中，您可能指的是c1，而不是c。

你可以这样做（如果你坚持使用 realloc()）：

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

#define MAX 50

int main() {
  char *c1 = malloc(MAX * sizeof(char));
  if(!c1) {
    printf("malloc() failed\n");
    return -1;
  }
  fgets(c1, MAX, stdin);
  int size = strlen(c1);
  char* c2;
  if(c1[size - 1] == '\n') {
    c1[size - 1] = '\0';
    c2 = realloc(c1, (size) * sizeof(char));
  } else {
    c2 = realloc(c1, (size + 1) * sizeof(char));
  }
  if(!c2) {
    printf("realloc() failed\n");
    return -1;
  }
  c1 = NULL;
  printf("Input was: %s\n", c2);
  free(c2);
  return 0;
}

这里有一些cmets：

1. 你希望读取一个字符串，因此你应该使用char*，而不是 int*.

2. 通常，您不希望保留fgets() 读取的换行符， 因此我使用了c1[size - 1] = '\0';，它覆盖了它。但是，如果用户输入了允许的最大字符数，那么就没有空间了，因此我们检查换行符是否真的存储在我们的缓冲区中。

3. Do not cast what malloc() returns.

4. 在realloc() 中，您应该为 字符串，加上 NULL 终止符。这就是为什么它现在是size+1。但是，在我们覆盖换行符的情况下，没有必要这样做，因为我们已经将字符串的大小减少了 1，因此 size 就足够了。

5. 永远不要忘记释放内存。

---

但是，如果我是你，我会打印一条消息，警告用户输入 MAX 字符并避免重新分配。

你看，正如ref of fgets() 所说，该函数将：

从流中读取字符并将它们作为 C 字符串存储到 str 中，直到读取 (num-1) 个字符或到达换行符或文件结尾，以先发生者为准。

所以，即使用户输入更多，程序也不会崩溃。

---

正如 Paul 正确评论的那样，使用第二个指针是一个非常好的主意，这样如果 realloc() 失败，我们就不会丢失数据。

还请注意，您不应释放c1，因为此指针已失效，应设置为NULL。 c1 可能会指向c2 指向的位置，因此如果我们释放c1，那么c2 将指向垃圾。或者可能是整个内存块被转移到其他地方，因此c1 指向垃圾，我们正在尝试释放。