-
Notifications
You must be signed in to change notification settings - Fork 1
/
Copy pathClone a linked list with next and random pointer-Optimised Approach
68 lines (60 loc) · 1.91 KB
/
Clone a linked list with next and random pointer-Optimised Approach
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
class Solution
{
private:
void insertAtTail(Node* &head, Node* &tail, int d){
Node* newNode = new Node(d);
if(head == NULL){
head = newNode;
tail = newNode;
return;
}
else{
tail->next = newNode;
tail = newNode;
}
}
public:
Node *copyList(Node *head)
{
//step 1: create a clone list
Node* cloneHead = NULL;
Node* cloneTail = NULL;
Node* temp = head;
while(temp!=NULL){
insertAtTail(cloneHead,cloneTail,temp->data);
temp = temp->next;
}
//step 2: clone nodes add between original List
Node* originalNode = head;
Node* cloneNode = cloneHead;
while(originalNode!= NULL && cloneNode!=NULL){
Node* next = originalNode->next;
originalNode->next = cloneNode;
originalNode = next;
next = cloneNode->next;
cloneNode->next = originalNode;
cloneNode = next;
}
//step 3: random pointer copy
Node* curr = head;
while(curr != NULL){
if(curr->next != NULL){
curr->next->arb = curr->arb ? curr->arb->next:curr->arb ;
//after wirting above line you dont have to write the below 4 lines
}curr = curr->next->next;
}
//step 4: revert changes done in step 2;
originalNode = head;
cloneNode = cloneHead;
while(originalNode != NULL && cloneNode != NULL){
originalNode->next = cloneNode->next;
originalNode = originalNode->next;
if(originalNode!=NULL){
cloneNode->next = originalNode->next;
}
cloneNode = cloneNode->next;
}
//step 5: return ans
return cloneHead;
}
};