题目
题目链接:138. 复制带随机指针的链表
思路–迭代
思路:
此题最难的不是复制链表结点,而是如何处理这个随机指针!
假如直接先把原链表的结点都直接复制到新链表中,那么此时就无法链接到随机指针;
所以我们必须想法个办法,在复制结点到新链表之前,要把随机指针给连接上;
我们知道,对于单链表我们是可以操作链表的任意位置的结点的,也就是说,当我在链表的某一cur结点的时候,我们可以在遍历链表时候记录curPrev结点,也就是得到cur之前的一个结点,我们也可以直接得到cur->next结点,也就是说,当我们知道单链表的某一个结点时候,我们是有办法拿到这个结点的前驱和后继结点!
那我们就有办法了,我们可以先把复制结点插入到与原链表的值相同的后面,那么这样,我们就有办法找到复制结点的前驱,由于前驱结点有链接随机指针的关系,那么我们就可以给复制链表的结点,也找到随机指针!
这是插入到复制结点后的样子
那么我们看橙色的7结点,也就是复制的结点,它可以通过紫色的7找到它的随机指针,那么橙色的7的随机结点就可以指向紫色7的随机结点,这就建立了复制结点和原链表结点的随机指针的关系
建立关系后:我们再定义一个新的头指针,把复制结点再拿下来尾插到新的头指针,同时连接上原来链表的结点,返回新头指针就完成了
/**
* Definition for a Node.
* struct Node {
* int val;
* struct Node *next;
* struct Node *random;
* };
*/
struct Node* copyRandomList(struct Node* head) {
struct Node* cur = head;
//复制结点到相同值到原链表相同结点值的后面
while(cur)
{
struct Node* copyNode = (struct Node* )malloc(sizeof(struct Node));
copyNode->val = cur->val;
//连接
copyNode->next = cur->next;
cur->next = copyNode;
//迭代
cur = copyNode->next;
}
//开始建立复制结点的随机指针关系
cur = head;
while(cur)
{
struct Node* copyNode = cur->next;
if(cur->random == NULL) //单独处理随机指针是空的情况
copyNode->random = NULL;
else //建立随机结点关系
copyNode->random = cur->random->next;
//迭代往前走
cur = copyNode->next;
}
//开始创建新链表,将复制结点拿下来,其实也是尾插的工作
struct Node* copyHead = NULL;
struct Node* copyTail = NULL;
cur = head;
while(cur)
{
struct Node* copyNode= cur->next;
struct Node* copyNodeNext = copyNode->next;
//开始头插到新的copyHead头指针中
if(copyTail == NULL)
{
copyHead = copyTail = copyNode;
}
else
{
copyTail->next = copyNode;
copyTail = copyNode;
}
//还原原来链表的链接关系
cur->next = copyNodeNext;
//cur迭代
cur = copyNodeNext;
}
return copyHead;
}