前言
最近做了一道非常有趣的链表题,题目是随机链表的复制,题中的链表与普通链表相比多了一个指向NULL或随机节点的随机指针,而这道题的解法非常妙。
一、问题分析
如题中描述的,我们需要拷贝这样一个包含随机指针的链表。明显这道题的难点主要集中在random指针的复刻,对于节点中的val与next都可以简单的遍历赋值。因此对于random的赋值,我们可以通过先将拷贝的新链表的每个节点放到原链表的各节点后,如下图所示:
然后对于新链表各节点的random便会是上一个节点的random的next节点,上个节点的random为NULL,则该节点的random也为NULL,最后再将两链表分离,便可以得到原链表的拷贝链表。
二、代码实现
typedef struct Node Node;
struct Node* copyRandomList(struct Node* head) {
if (head == NULL)
return NULL;
Node* pcur = head;
while (pcur)//将新链表的各个插到对应原链表节点后
{
Node* copy = (Node*)malloc(sizeof(Node));
copy->next = pcur->next;
copy->val = pcur->val;
pcur->next = copy;
pcur = copy->next;
}
pcur = head;
while (pcur)//将新节点的random连接
{
Node* copy = pcur->next;
if (pcur->random == NULL)
copy->random = NULL;
else
copy->random = pcur->random->next;
pcur = copy->next;
}
pcur = head;
Node* phead, *ptail;
ptail = phead = pcur->next;
while (pcur)//将原链表与新链表分割
{
pcur->next = ptail->next;
pcur = pcur->next;
if (pcur)
{
ptail->next = pcur->next;
ptail = ptail->next;
}
}
return phead;
}