先看一段代码:
#include<iostream>
using namespace std;
void GetMe(char *p,int num)
{
p=(char*)malloc(sizeof(char)*num);
}
int main()
{
char *str=NULL;
GetMe(str,100);
strcpy(str,"Hello!");
cout<<str<<endl;
return 0;
}
该代码运行崩溃,原因是str并没有实际获得指针p开辟的内存空间。在函数GetMe中,p实际上是主函数中str的一个副本,p申请了内存,只是把p指向的内存地址改变,而str并没有改变,所以str依然没有获得内存,在执行字符串复制时就会出错。而且每次p申请的内存都不会得到释放,最终会造成内存泄露。
我们可以更改以上代码如下:
#include<iostream>
using namespace std;
char* GetMe(char *p,int num)
{
p=(char*)malloc(sizeof(char)*num);
return p;
}
int main()
{
char *str=NULL;
str=GetMe(str,100);
strcpy(str,"Hello!");
cout<<str<<endl;
delete str;
return 0;
}
在该代码中我们为p申请了空间,并将该空间的地址作为返回值传给str,这样str就指向了p申请的内存空间。还有一种修改方法是:传递str的指针:
#include<iostream>
using namespace std;
void GetMe(char **p,int num)
{
*p=(char*)malloc(sizeof(char)*num);
}
int main()
{
char *str=NULL;
GetMe(&str,100);
strcpy(str,"Hello!");
cout<<str<<endl;
delete str;
return 0;
}
在该代码中传递了str的指针给函数GetMe(),那么p就是str的地址的副本,地址的副本可定就是该地址了,所以该函数是为str地址指向的str开辟空间。所以能正确输出结果。