题目
/** * Definition for a Node. * struct Node { * int val; * struct Node *next; * struct Node *random; * }; */ struct Node* copyRandomList(struct Node* head) { }
解题思路
思路一
要复制一个链表是简单的,但这个是带有随机指针的链表,你不能确定单节点的随机指针指向哪个,这就需要动一下脑筋了。可以发现这本身是一个单链表,即链表是单向链接的,随机指针指向的节点也都是单链表内的一个节点,所以我们可以采取一个数组来记忆每个单向节点中的random是指向第几个节点(这样同时可以避免一个问题,用val值标记的话如第二个节点和第五个节点(val 均为1),就难以分辨val=11节点的random应指向哪里,而采用顺序表进行存储random指向可以解决此问题)
如何实现?
如找到第三个节点(val=11)的random指向,就要采用嵌套循环,一个循环遍历链表random指向,一个循环存储到数组中对应位置。如图示:
但这样会有一个问题,时间复杂度为O(N^2), 要优化到O(N),就不能有两个嵌套循环。
思路2
如果我们复制一遍原单向链表,将新单链表与原单链表进行链接,是不是就可以通过仅遍历一遍将原单链表的random指针指向复制到新单链表上呢? 通过画图可以发现是可行的:
//拷贝节点链接在源节点后面 struct Node* cur = head; while(cur) { struct Node* copy = (struct Node*)malloc(sizeof(struct Node)); struct Node* next = cur->next; copy->val = cur->val; //进行链接 cur ->next = copy; copy->next = next; cur = next; }
如何链接拷贝节点的random?
我们用cur指针对原链表进行遍历,可以发现(如val =13 的copy节点),当cur指向val= 13 节点时,cur ->random 指向节点的下一个就是val= 7 的copy节点,也就是说 copy ->random = cur->random->next;
对剩余节点同样是适用的,当cur == NULL 时遍历结束, 同时应主要判断cur ->random 是否指向NULL和cur是否重置为头结点head,以便进行新一次遍历
// 链接拷贝节点的random cur = head; // 对cur 重置 while(cur) { struct Node*copy = cur ->next; struct Node* next = copy ->next; //设置next 指针,便于观察 if (cur ->random == NULL) { copy->random = NULL; } else { copy->random = cur->random->next; } cur = next; }
最后一步就是将拷贝节点拆解下来,进行链接成拷贝链表,同时可以把原链表恢复链接关系(不恢复OJ也能过),如何实现?只需利用尾插法再用cur指针遍历一遍,所以我们需要设置copyHead 指针(用于返回头节点)和copyTail指针用于尾插进行遍历
//拆解拷贝节点,链接组成拷贝链表 cur = head; struct Node* copyHead = NULL, *copyTail = NULL; while(cur) { struct Node* copy = cur ->next; struct Node* next = copy->next; //链接原链表 cur->next = next; //尾插拷贝链表 if(copyTail == NULL) { copyHead = copyTail = copy; } else { copyTail-> next = copy; //此处为下一个copy节点,如val=7的节点 下一个是val=13 copyTail = copyTail ->next; } cur = next; } return copyHead; }
本题利用了链表尾插,删除等思想,对指针掌握能力要求较高
这是题解:
struct Node* copyRandomList(struct Node* head) { //拷贝节点链接在源节点后面 struct Node* cur = head; while(cur) { struct Node* copy = (struct Node*)malloc(sizeof(struct Node)); struct Node* next = cur->next; copy->val = cur->val; cur ->next = copy; copy->next = next; cur = next; } // 链接拷贝节点的random cur = head; while(cur) { struct Node*copy = cur ->next; struct Node* next = copy ->next; if (cur ->random == NULL) { copy->random = NULL; } else { copy->random = cur->random->next; } cur = next; } //拆解拷贝节点,链接组成拷贝链表 cur = head; struct Node* copyHead = NULL, *copyTail = NULL; while(cur) { struct Node* copy = cur ->next; struct Node* next = copy->next; //链接原链表 cur->next = next; //尾插拷贝链表 if(copyTail == NULL) { copyHead = copyTail = copy; } else { copyTail-> next = copy; //此处为下一个copy节点,如val=7的节点 下一个是val=13 copyTail = copyTail ->next; } cur = next; } return copyHead; }
本节完,有不懂可以私聊作者