/**
* 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) {}
* };
*/
//from anniekim
class Solution {
public:
public:
RandomListNode *copyRandomList(RandomListNode *head) {
return copyRandomList_1(head);
}
RandomListNode *copyRandomList_1(RandomListNode *head) {
for (RandomListNode *cur = head; cur; cur = cur->next->next) {
RandomListNode *newNode = new RandomListNode(cur->label);
newNode->next = cur->next;
cur->next = newNode;
}
for (RandomListNode *cur = head; cur; cur = cur->next->next)
if (cur->random)
cur->next->random = cur->random->next;
RandomListNode dummy(0), *curNew = &dummy;
for (RandomListNode *cur = head; cur; cur = cur->next) {
curNew->next = cur->next;
curNew = curNew->next;
cur->next = cur->next->next;
}
return dummy.next;
}
RandomListNode *copyRandomList_2(RandomListNode *head) {
if (!head) return NULL;
unordered_map<RandomListNode *, RandomListNode *> map;
RandomListNode dummy(0), *curNew = &dummy, *cur = head;
while (cur)
{
if (map.find(cur) == map.end())
map[cur] = new RandomListNode(cur->label);
if (cur->random && map.find(cur->random) == map.end())
map[cur->random] = new RandomListNode(cur->random->label);
curNew->next = map[cur];
curNew = curNew->next;
curNew->random = map[cur->random];
cur = cur->next;
}
return dummy.next;
}
};
No comments:
Post a Comment