指针传递动态内存

指针传递动态内存

问题描述

代码如下，本意是想传入一个头指针，生成一个链表，但是结果返回的head仍然是空。

int main(){
    DataType data[5] = {1, 4, 5, 3, 2};
    LinkNode * link1;
    LinkNode_create(link1, data, 5);
    return 0;
}

int LinkNode_create(LinkNode * head, DataType datas[], int sum){

    int i = 0;
    LinkNode * p, *q;
    head = (LinkNode*)malloc(sizeof(LinkNode));

    if(head){
            head->value = 0;
            head->next = NULL;
    }else{
            printf("No enough memory\n");
            return ERROR;
    }

    q = head;
    for(i = 0; i < sum; i++){
            p = (LinkNode*)malloc(sizeof(LinkNode));
            p->value = datas[i];
            p->next = NULL;

                    q->next = p;
            q = q->next;
    }

    return SUCC;
}

原因

解释如下：
在main中，实参link1 = NULL;
调用 LinkNode_create后，link1的值NULL被赋给了head,即此时

link1 = NULL;
head = NULL;

进入LinkNode_create:
head指向了一个链表，但是函数调用完成后，局部变量被收回，head不存在了。

也就是说，现在内存中有了一个链表，但是link1并没有指向它。

解决方案

想要解决这个问题，有两种方法
1. 将head作为返回值
2. 将link1的地址传入

方法一 ： 返回头指针

方法一比较简单，也比较好理解，修改后代码如下：

int main(){
    DataType data[5] = {1, 4, 5, 3, 2};
    LinkNode * link1 = LinkNode_create( data, 5);  // return
    return 0;
}

LinkNode* LinkNode_create( DataType datas[], int sum){  // return
    LinkNode * head;
    int i = 0;
    LinkNode * p, *q;
    head = (LinkNode*)malloc(sizeof(LinkNode));

    if(head){
            head->value = 0;
            head->next = NULL;
    }else{
            printf("No enough memory\n");
    }

    q = head;
    for(i = 0; i < sum; i++){
            p = (LinkNode*)malloc(sizeof(LinkNode));
            p->value = datas[i];
            p->next = NULL;

                    q->next = p;
            q = q->next;
    }

    return head;   //return 
}

方法二：指针的指针

想要link1 指向一个链表的头结点；
将link1的地址 link1_pointer 传给 LinkNode_create();
在LinkNode_create()中，head_pointer获得了link1的地址；

通过
*head_pointer = (LinkNode*)malloc(sizeof(LinkNode)); // allocate memory
创建的链表的头指针的地址放到*head_pointer（即link1）中，这样link1就指向了创建的链表的头结点；
为了便于阅读，将这个值再赋给一个名为head的指针：
‘head = *head_pointer; // this sentence must be after the previous one’

NOTE:
*head_pointer = (LinkNode*)malloc(sizeof(LinkNode));必须要放在head = *head_pointer;前面，如果次序反了，那么就会像这样：
head = *head_pointer;
此时*head_pointer与link1相等，在main()中，link1定义但未初始化，指向一个脏数据，所以head也会指向这个脏数据，所以head现在是一个野指针。
* head_pointer = (LinkNode*)malloc(sizeof(LinkNode));
现在link1指向了一个新创建的链表的头结点，但是head仍然是一个野指针。这样就无法使用head清晰明了地进行接下来的步骤。

完整代码如下：

int main(){
    DataType data[5] = {1, 4, 5, 3, 2};
    LinkNode * link1;
    LinkNode ** link1_pointer = &link1;  // get the pointer of link1
    LinkNode_create(link1, data, 5);
    return 0;
}

int LinkNode_create(LinkNode ** head_pointer, DataType datas[], int sum){

    int i = 0;
    LinkNode * head,
    LinkNode * p, *q;
    *head_pointer = (LinkNode*)malloc(sizeof(LinkNode));  // allocate memory
    head = *head_pointer;   // this sentence must be after the previous one 
    if(head){
            head->value = 0;
            head->next = NULL;
    }else{
            printf("No enough memory\n");
            return ERROR;
    }

    q = head;
    for(i = 0; i < sum; i++){
            p = (LinkNode*)malloc(sizeof(LinkNode));
            p->value = datas[i];
            p->next = NULL;

                    q->next = p;
            q = q->next;
    }

    return SUCC;
}