-
Notifications
You must be signed in to change notification settings - Fork 3
/
Copy path03_reorder_list.py
85 lines (60 loc) · 1.65 KB
/
03_reorder_list.py
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
class Node:
def __init__(self, data):
self.data = data
self.next = None
class SinglyLinkedList:
def __init__(self):
self.head = None
def append(self, data):
new_node = Node(data)
if self.head is None:
self.head = new_node
return
last_node = self.head
while last_node.next:
last_node = last_node.next
last_node.next = new_node
def print_list(self):
cur_node = self.head
while cur_node:
print(cur_node.data)
cur_node = cur_node.next
def reorderList(self):
slow, fast = self.head, self.head.next
while fast and fast.next:
slow = slow.next
fast = fast.next.next
second = slow.next
prev = slow.next = None
while second:
tmp = second.next
second.next = prev
prev = second
second = tmp
# merge two halfs
first, second = self.head, prev
while second:
tmp1, tmp2 = first.next, second.next
first.next = second
second.next = tmp1
first, second = tmp1, tmp2
if __name__ == "__main__":
llist1 = SinglyLinkedList()
llist1.append(1)
llist1.append(2)
llist1.append(3)
llist1.append(4)
llist1.print_list()
print("\n")
llist1.reorderList()
llist1.print_list()
llist2 = SinglyLinkedList()
llist2.append(1)
llist2.append(2)
llist2.append(3)
llist2.append(4)
llist2.append(5)
llist2.print_list()
print("\n")
llist2.reorderList()
llist2.print_list()