面试题:写一个函数实现一个:(分配指定大小的内存块,可以使用malloc)
int my_malloc(char **ptr,int size)
使用下面的形式来用这个函数:
*ptr=NULL;
my_malloc(&ptr,10)
正确解法:
int my_malloc(char** ptr, int size) {
char* temp = NULL;
//ptr应该指向传递进来的指针,pt
if (ptr == NULL) {
return -1;
}
//内存的申请
temp = (char*)malloc(sizeof(char) * size);
if (temp == NULL) {
return -1;
}
*ptr = temp;
}
int main()
{
//注意,要传入的是&pt,让二级指针指向pt
char* pt = NULL;
my_malloc(&pt,10);
memcpy(pt, "hello",5);
for (int i = 0; i < 5; i++) {
cout << *pt;
pt++;
}
return 0;
}
解析:
第一步:my_malloc(pt); 执行 ptr = &pt;让二级指针ptr指向 一级指针 pt;
第二步:malloc()函数分配内存
第三步:一级指针 temp 指向刚才malloc()分配的内存空间
第四步:*ptr = temp; 由于 *p = pt; 故 实际执行语句 pt= temp( 此时pt已经指向malloc()分配的内存)
第五步:函数执行完毕,释放二级指针ptr和一级指针temp ,malloc()分配的内存空间由于在堆区而被保留了下来
此时pt已经指向malloc()分配的内存。使用函数用二级指针间接分配内存成功。
为什么一级指针不行呢?
错误解法:
int my_malloc(char* ptr, int size) {
char* temp = NULL;
//ptr应该指向传递进来的指针,pt
if (ptr == NULL) {
return -1;
}
//内存的申请
temp = (char*)malloc(sizeof(char) * size);
if (temp == NULL) {
return -1;
}
ptr = temp;
}
int main()
{
//注意,要传入的是&pt,让二级指针指向pt
char* pt = NULL;
my_malloc(&pt,10);
memcpy(pt, "hello",5);
for (int i = 0; i < 5; i++) {
cout << *pt;
pt++;
}
return 0;
}
解析:
第一步:my_malloc(pt); 执行 ptr = pt;让一级指针ptr指向 一级指针 pt;
第二步:malloc()函数分配内存
第三步:一级指针 temp 指向刚才malloc()分配的内存空间
第四步:ptr = tmp; 此时ptr 的指向关系被改变,由原来的指向pt变为指向temp;
( 此时pt仍然指向NULL)