题目
思路
难点在于复制链表的random域不能指向原链表的节点,必须指向赋值链表的节点。所以我们需要将通过原链表的相对位置来确定复制链表的相对位置,一个思路就是将原链表中的每个节点后面加上复制节点,这样复制节点的random域就是原节点random域的next所指向的地址(random域指向NULL需要特殊处理)
接下来思路很清晰了
1.将原链表中每个节点尾插复制节点
2. 通过原链表节点确定复制链表节点的random域
3. 将复制链表取下来
代码
/**
* Definition for a Node.
* struct Node {
* int val;
* struct Node *next;
* struct Node *random;
* };
*/
struct Node* copyRandomList(struct Node* head) {
typedef struct Node Node;
if (head == NULL) return NULL;
//1. 将每个复制节点插入到原节点之后
Node* cur = head;
while (cur)
{
Node* copy = (Node*)malloc(sizeof(Node));//这里分配一个指针的大小就可以了
copy->next = cur->next;
copy->val = cur->val;//复制值
cur->next = copy;
cur = copy->next;
}
//2. 通过原节点设置复制节点random
cur = head;
while (cur)
{
Node* copy = cur->next;
if (cur->random != NULL)
copy->random = cur->random->next;
else
copy->random = NULL;
cur = copy->next;
}
//3. 切断copy链表
Node* copyHead = head->next;//记录copy链表头
cur = head;
while (cur)
{
Node* copy = cur->next;
Node* tmp = copy->next;//临时记录复制节点的下一个原节点
if (tmp != NULL)
copy->next = tmp->next;
cur->next = tmp;
//cur指向下一个原节点
cur = tmp;
}
return copyHead;
}