-
Notifications
You must be signed in to change notification settings - Fork 2
/
Copy path430. Flatten a Multilevel Doubly Linked List.cpp
177 lines (126 loc) · 3.82 KB
/
430. Flatten a Multilevel Doubly Linked List.cpp
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
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
You are given a doubly linked list, which contains nodes that have a next pointer,
a previous pointer, and an additional child pointer.
This child pointer may or may not point to a separate doubly linked list,
also containing these special nodes.
These child lists may have one or more children of their own, and so on,
to produce a multilevel data structure as shown in the example below.
Given the head of the first level of the list, flatten the list so that all the nodes
appear in a single-level, doubly linked list. Let curr be a node with a child list.
The nodes in the child list should appear after curr and before curr.next in the flattened list.
Return the head of the flattened list.
The nodes in the list must have all of their child pointers set to null.
How the multilevel linked list is represented in test cases:
We use the multilevel linked list from Example 1 above:
1---2---3---4---5---6--NULL
|
7---8---9---10--NULL
|
11--12--NULL
The serialization of each level is as follows:
[1,2,3,4,5,6,null]
[7,8,9,10,null]
[11,12,null]
To serialize all levels together, we will add nulls in each level to signify
no node connects to the upper node of the previous level.
The serialization becomes:
[1, 2, 3, 4, 5, 6, null]
|
[null, null, 7, 8, 9, 10, null]
|
[ null, 11, 12, null]
Merging the serialization of each level and removing trailing nulls we obtain:
[1,2,3,4,5,6,null,null,null,7,8,9,10,null,null,11,12]
Example 1:
Input: head = [1,2,3,4,5,6,null,null,null,7,8,9,10,null,null,11,12]
Output: [1,2,3,7,8,11,12,9,10,4,5,6]
Example 2:
Input: head = [1,2,null,3]
Output: [1,3,2]
Example 3:
Input: head = []
Output: []
Constraints:
The number of Nodes will not exceed 1000.
1 <= Node.val <= 10^5
********************************************************************************
# Approach 1:
/*
// Definition for a Node.
class Node {
public:
int val;
Node* prev;
Node* next;
Node* child;
};
*/
class Solution {
public:
Node* flatten(Node* head) {
if (!head) return head;
stack<Node*> s;
Node *curr = head, *tail = NULL;
while (curr) {
if (curr->child) {
if (curr->next) s.push(curr->next);
Node *child = curr->child;
curr->child = NULL;
curr->next = child;
child->prev = curr;
}
tail = curr;
curr = curr->next;
}
while (!s.empty()) {
curr = s.top();
s.pop();
curr->prev = tail;
tail->next = curr;
while (curr) {
tail = curr;
curr = curr->next;
}
}
return head;
}
};
TC -> O(n), n is the length of the linked list
SC -> O(k), k is the number of levels in the linked list
********************************************************************************
# Approach 2:
/*
// Definition for a Node.
class Node {
public:
int val;
Node* prev;
Node* next;
Node* child;
};
*/
class Solution {
public:
Node* flatten(Node* head) {
if (!head) return head;
Node *curr = head, *last = NULL;
while (curr) {
if (curr->child) {
Node *ahead = curr->next;
Node *child = curr->child;
curr->child = NULL;
curr->next = child;
child->prev = curr;
last = child;
while (last->next) last = last->next;
if (ahead) {
last->next = ahead;
ahead->prev = last;
}
}
curr = curr->next;
}
return head;
}
};
TC -> O(n), n is the length of the linked list
SC -> O(1)