C編程：如何在此程序中使用realloc？

Question

我必須編寫一個程序來存儲和打印出內存中的整數。我必須使用realloc。基本上，該程序分配2個整數的大小。當輸入給出2個整數時，它應該重新分配1個int空間並打印出double。接下來，當輸入端爲3個整數，它應該分配爲INT 2個空格，並打印出雙..等..

Test cases: 

input file in.0: 
------ 
4 
------ 

expected output: 
------ 
4 
------ 

================================================= 

input file in.1: 
------ 
4 5 
------ 

expected output: 
------ 
4 
5 
double 
------ 

================================================== 

input file in.2: 
------ 
4 5 3 
------ 

expected output: 
------ 
4 
5 
double 
3 
double 
------ 

=================================================== 

input file in.3: 
------ 
4 5 3 2 9 
------ 

expected output: 
------ 
4 
5 
double 
3 
double 
2 
9 
double 

我寫了這個程序，但它不能正確分配內存。有人能指導我寫作方向嗎？

int main(void) 
{ 
    int c; 

    int digit; 
    int count = 0; 
    int d_size = 1; 
    int init_size = 2; 
    int *p = (int *) malloc(sizeof(int) * init_size); 

    while((c = scanf("%i", &digit)) != EOF) 
    { 
     if (c == 1) 
     { 
      *(p+count) = digit; 
      count++; 
     } 
     else 
     { 
      printf("not valid"); 
     } 

     printf("%i\n", digit); 

     if (count >= 2) 
     { 
      printf("double\n"); 
      p = (int *) realloc(p, sizeof(int) * d_size); 
      d_size = d_size * 2; 
     } 

    } 

Answer 1

你init_size是2，但你的d_size是1.首先，使d_size等於init_size。其次，你需要在realloc之前做d_size = d_size * 2，這樣你才能真正增加尺寸。

---

備註：realloc如果內存不足，將會失敗。如果你寫：

p = realloc(p, ...); 

如果失敗，你將失去先前分配的內存。你應該總是使用realloc這樣的：

enlarged = realloc(p, ...); 
if (enlarged == NULL) 
    // handle error 
else 
    p = enlarged; 

---

邊注2：你可能最終會改變你的指針的類型。最好不要重複。取而代之的

int *p; 
p = (int *)malloc(sizeof(int) * count); 

寫：

int *p; 
p = malloc(sizeof(*p) * count);