自动扩容版 gets

永恒的蓝色梦想

1. #include<stdio.h>
   
2. #include<stdlib.h>
   
3. #include<malloc.h>
   
4. 
   
5. 
   
6. 
   
7. #ifndef DEFAULT_STRING_LENGTH
   
8. #define DEFAULT_STRING_LENGTH 50 //默认长度 50
   
9. #endif
   
10. 
    
11. 
    
12. 
    
13. //自动扩容版的 fgets
    
14. char* autoalloc_fgets(char* _Str, size_t* const _Capacity, FILE* const _Stream) {
    
15.     /*
    
16.     警告：
    
17.         只能用于动态分配的数组，静态数组使用后果自负。
    
18.     */
    
19.     char* _Temp;
    
20.     size_t _Index = 0;
    
21. 
    
22. 
    
23.     //输入检查
    
24.     if (!*_Capacity) {//容量为 0
    
25.         _Str = (char*)malloc(DEFAULT_STRING_LENGTH * sizeof(char));//分配 DEFAULT_STRING_LENGTH 长度的空间
    
26. 
    
27. 
    
28.         if (_Str) {//分配成功
    
29.             *_Capacity = DEFAULT_STRING_LENGTH;
    
30.         }
    
31.         else {//分配失败
    
32.             return NULL;
    
33.         }
    
34.     }
    
35.     else if (!_Str) {//传入空指针
    
36.         _Str = (char*)malloc(*_Capacity * sizeof(char));//分配* _Capacity 长度的空间
    
37. 
    
38.         if (!_Str) {//分配失败
    
39.             *_Capacity = 0;
    
40.             return NULL;
    
41.         }
    
42.     }
    
43. 
    
44. 
    
45. #define _Len (_Index + 1)
    
46.     for (;;) {
    
47.         if (_Len < *_Capacity) {
    
48.         LABEL:
    
49.             switch (_Str[_Index] = fgetc(_Stream)) {//遇到以下字符，终止扫描
    
50.                 case '\n':
    
51.                 case '\r':
    
52.                 case EOF:
    
53.                     _Str[_Index] = '\0';//'\0' 的话，就不需要重置了
    
54. 
    
55.                 case '\0':
    
56.                     *_Capacity = _Len;
    
57.                     return (char*)realloc(_Str, *_Capacity * sizeof(char));
    
58.             }
    
59. 
    
60.             ++_Index;
    
61.         }
    
62.         else {//扩容，默认扩容为原来的两倍
    
63.             *_Capacity <<= 1;
    
64.             if (_Temp = (char*)realloc(_Str, *_Capacity * sizeof(char))) {//为了防止 realloc 返回 NULL 造成内存泄露
    
65.                 _Str = _Temp;
    
66.                 goto LABEL;//扩容完之后不需要重新判断，goto 提高效率
    
67.             }
    
68.             else {//扩容失败
    
69.                 _Str[_Index] = '\0';//放置字符串终止符
    
70.                 *_Capacity >>= 1;//退回来
    
71.                 return _Str;
    
72.             }
    
73.         }
    
74.     }
    
75. #undef _Len
    
76. 
    
77. 
    
78.     return NULL;//装饰
    
79. }
    
80. 
    
81. 
    
82. 
    
83. //自动扩容版的 gets
    
84. char* autoalloc_gets(char* _Str, size_t* const _Capacity) {
    
85.     return autoalloc_fgets(_Str, _Capacity, stdin);//直接调用 autoalloc_fgets
    
86. }

复制代码

weiter

谢谢

Cool_Breeze

重新开辟内存空间，再把原来的内容复制过怎么样？

永恒的蓝色梦想

Cool_Breeze 发表于 2020-8-15 15:33
重新开辟内存空间，再把原来的内容复制过怎么样？

？