-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathlinkedlist.py
More file actions
186 lines (150 loc) · 4.35 KB
/
linkedlist.py
File metadata and controls
186 lines (150 loc) · 4.35 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
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
178
179
180
181
182
183
184
185
186
class Node(object):
def __init__(self, value=None, Next=None):
self.value, self.next = value, Next
def __str__(self):
return "<Node: value:{}, next:{} >".format(self.value, self.next)
__repr__ = __str__
class LinkedList(object):
def __init__(self, maxsize=None):
self.root = Node()
self.tailnode = None
self.maxsize = maxsize
self.length = 0
def __len__(self):
return self.length
def append(self, value):
if self.maxsize is not None and self.length >= self.maxsize:
raise Exception('Full')
node = Node(value)
if self.tailnode is None:
self.root.next = node
else:
tailnode = self.tailnode
tailnode.next = node
self.tailnode = node
self.length += 1
def appendleft(self, value):
if self.maxsize is not None and self.length >= self.maxsize:
raise Exception('Full')
node = Node(value)
if self.tailnode is None:
self.root.next = node
self.tailnode = node
headnode = self.root.next
node.next = headnode
self.root.next = node
self.length += 1
def __iter__(self):
for node in self._iter_node():
yield node.value
def _iter_node(self):
curnode = self.root.next
while curnode is not self.tailnode: # 从第一个节点开始遍历
yield curnode
curnode = curnode.next # 移动到下一个节点
if curnode is not None:
yield curnode
def find(self, value):
index = 0
for node in self._iter_node():
if node.value == value:
return index
index += 1
return -1
def remove(self, value):
prevnode = self.root
for curnode in self._iter_node():
if curnode.value == value:
prevnode.next = curnode.next
self.length -= 1
if curnode is self.tailnode:
self.tailnode = prevnode
del curnode
return 1
prevnode = curnode
return -1
def popleft(self):
if self.root.next is None:
raise Exception('The LinkedList is Empty')
headnode = self.root.next
self.root.next = headnode.next
value = headnode.value
if headnode is self.tailnode:
self.tailnode = None
self.length -= 1
del headnode
return value
def clear(self):
for node in self._iter_node():
del node
self.length = 0
self.root.next = None
self.tailnode = None
def reverse(self):
curnode = self.root.next
self.tailnode = curnode
prevnode = None
while curnode:
nextnode = curnode.next
curnode.next = prevnode
if nextnode is None:
self.root.next = curnode
prevnode = curnode
curnode = nextnode
def test_linked_list():
ll = LinkedList()
ll.append(0)
ll.append(1)
ll.append(2)
ll.append(3)
assert len(ll) == 4
assert ll.find(2) == 2
assert ll.find(-1) == -1
assert ll.remove(0) == 1
assert ll.remove(10) == -1
assert ll.remove(2) == 1
assert len(ll) == 2
assert list(ll) == [1, 3]
assert ll.find(0) == -1
ll.appendleft(0)
assert list(ll) == [0, 1, 3]
assert len(ll) == 3
headvalue = ll.popleft()
assert headvalue == 0
assert len(ll) == 2
assert list(ll) == [1, 3]
assert ll.popleft() == 1
assert list(ll) == [3]
ll.popleft()
assert len(ll) == 0
assert ll.tailnode is None
ll.clear()
assert len(ll) == 0
assert list(ll) == []
def test_linked_list_remove():
ll = LinkedList()
ll.append(3)
ll.append(4)
ll.append(5)
ll.append(6)
ll.append(7)
ll.remove(7)
print(list(ll))
def test_linked_list_reverse():
ll = LinkedList()
n = 10
for i in range(n):
ll.append(i)
# print(list(ll))
ll.reverse()
# print(list(ll))
assert list(ll) == list(reversed(range(n)))
def test_linked_list_append():
ll = LinkedList()
ll.appendleft(1)
ll.append(2)
assert list(ll) == [1, 2]
if __name__ == '__main__':
test_linked_list()
test_linked_list_append()
test_linked_list_reverse()