-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathe.py
More file actions
80 lines (56 loc) · 1.94 KB
/
Copy pathe.py
File metadata and controls
80 lines (56 loc) · 1.94 KB
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
from __future__ import annotations
import os
LOCAL = os.environ.get('REMOTE_JUDGE', 'false') != 'true'
if LOCAL:
class DoubleConnectedNode:
def __init__(self,
value: str,
next: DoubleConnectedNode | None = None,
prev: DoubleConnectedNode | None = None) -> None:
self.value = value
self.next = next
self.prev = prev
def solution(node: DoubleConnectedNode) -> DoubleConnectedNode:
return reverse_linked_list(node)
def reverse_linked_list(node: DoubleConnectedNode) -> DoubleConnectedNode:
linked_list = LinkedList(node)
linked_list.reverse()
return linked_list.get_head()
class LinkedList:
head: DoubleConnectedNode
def __init__(self, head: DoubleConnectedNode) -> None:
self.head = head
def get_head(self) -> DoubleConnectedNode:
return self.head
def reverse(self) -> None:
reversed_node = self.head
while True:
next_reversed_node = reversed_node.prev
previous_reversed_node = reversed_node.next
reversed_node.next = next_reversed_node
reversed_node.prev = previous_reversed_node
if previous_reversed_node is None:
break
reversed_node = previous_reversed_node
self.head = reversed_node
def test() -> None:
node3 = DoubleConnectedNode('node3')
node2 = DoubleConnectedNode('node2')
node1 = DoubleConnectedNode('node1')
node0 = DoubleConnectedNode('node0')
node0.next = node1
node1.prev = node0
node1.next = node2
node2.prev = node1
node2.next = node3
node3.prev = node2
new_head = solution(node0)
assert new_head is node3
assert node3.next is node2
assert node2.next is node1
assert node2.prev is node3
assert node1.next is node0
assert node1.prev is node2
assert node0.prev is node1
if __name__ == '__main__':
test()