-
Notifications
You must be signed in to change notification settings - Fork 1
Expand file tree
/
Copy pathflatten.cpp
More file actions
47 lines (41 loc) · 931 Bytes
/
flatten.cpp
File metadata and controls
47 lines (41 loc) · 931 Bytes
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
// Definition for a Node.
class Node {
public:
int val;
Node* prev;
Node* next;
Node* child;
};
class Solution {
public:
Node* flatten(Node* head)
{
Node *res = head;
Node *tmp, *tmp1, *tail;
while(res)
{
if(res->child != NULL)
{
tmp = res->next;
tmp1 = flatten(res->child);
res->child = NULL;
res->next = tmp1;
tmp1->prev = res;
tail = tmp1;
while(tail->next != NULL)
tail = tail->next;
tail->next = tmp;
if(tmp)
tmp->prev = tail;
}
res = res->next;
}
return head;
}
};
int main()
{
//create doubly linked list
Node *head = flatten(head);
//print linked list
}