leetcode - Copy List with Random Pointer
來源：程序員人生 發布時間：2014-10-08 09:31:53 閱讀次數：2919次
A linked list is given such that each node contains an additional random pointer which could point to any node in the list or null.
Return a deep copy of the list.
/**
* 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) {}
* };
*/
struct RandomListNode
{
int label;
RandomListNode *next,*random;
RandomListNode(int x) : label(x), next(NULL), random(NULL) {}
};
class Solution {
public:
RandomListNode *copyRandomList(RandomListNode *head) {
if(head == NULL) return head;
//將oldList的每一個節點之后，插入一個newNode.
RandomListNode *oldListNode = head;
while(oldListNode != NULL)
{
RandomListNode *newListNode = new RandomListNode(oldListNode->label);
newListNode->next = oldListNode->next;
newListNode->random = oldListNode->random;
oldListNode->next = newListNode;
oldListNode = oldListNode->next->next;
}
//update newListNode上的random結點關聯的結點
oldListNode = head;
while(oldListNode != NULL)
{
if(oldListNode->random != NULL)
{
oldListNode->next->random = oldListNode->random->next;
}
oldListNode = oldListNode->next->next;
}
//分離oldListNode與newListNode
RandomListNode *newListNode = new RandomListNode(0);
newListNode->next = head;
oldListNode = head;
RandomListNode *resultListNode = newListNode;
while(oldListNode != NULL)
{
newListNode->next = oldListNode->next;
oldListNode->next = newListNode->next->next;
newListNode = newListNode->next;
oldListNode = oldListNode->next;
}
return resultListNode->next;
}
};
生活不易，碼農辛苦
如果您覺得本網站對您的學習有所幫助,可以手機掃描二維碼進行捐贈
