给出一个链表,每个节点包含一个额外增加的随机指针可以指向链表中的任何节点或空的节点。
返回一个深拷贝的链表。
挑战
可否使用O(1)的空间
/**
* Definition for singly-linked list with a random pointer.
* struct RandomListNode {
* int label;
* RandomListNode *next, *random;
* RandomListNode(int x) : label(x), next(NULL), random(NULL) {}
* };
*/
class Solution {
public:
/**
* @param head: The head of linked list with a random pointer.
* @return: A new head of a deep copy of the list.
*/
RandomListNode *copyRandomList(RandomListNode *head) {
// write your code here
if (head == NULL)
{
return NULL;
}
RandomListNode *p = head;
RandomListNode *q = NULL;
while (p != NULL)
{
RandomListNode *next = p->next;
q = new RandomListNode(p->label);
p->next = q;
q->next = next;
q->random = p->random;
p = next;
}
RandomListNode *result = head->next;
p = head;
while (p != NULL)
{
if (p->next->random != NULL)
{
p->next->random = p->next->random->next;
}
p = p->next->next;
}
p = head;
while (p != NULL)
{
q = p->next;
p->next = p->next->next;
if (p->next == NULL)
{
q->next = NULL;
}
else
{
q->next = p->next->next;
}
p = p->next;
}
return result;
}
};