双端链表
class Node(object):
def __init__(self,value=None,prev=None,next=None):
self.value,self.prev,self.next = value,prev,next
class DoubleLinkedList(object):
def __init__(self,maxsieze=None):
self.maxsieze=maxsieze
node = Node()
node.next,node.prev = node,node
self.root = node
self.length = 0
def __len__(self):
return self.length
def headnode(self):
return self.root.next
def tailnode(self):
return self.root.prev
def append(self,value):
if self.maxsieze is not None and len(self) > self.maxsieze:
raise Exception("List is Full")
node = Node(value=value)
tailnode = self.tailnode()
tailnode.next = node
node.prev = tailnode
node.next = self.root
self.root.prev = node #互指
self.length += 1
def appendleft(self,value):
if self.maxsieze is not None and len(self) > self.maxsieze:
raise Exception("List is Full")
node = Node(value=value)
if self.root.next is self.root: #如果链表为空
node.next = self.root
node.prev = self.root
self.root.next = node
self.root.prev = node
else:
node.prev = self.root
headnode = self.root.next
node.next = headnode
headnode.prev = node
self.root.next = node
self.length += 1
def remove(self,node): #删除为O(1) 所以删除节点而不是值
if node is self.root:
return
else:
node.prev.next = node.next
node.next.prev = node.prev
self.length -= 1
return node
def iter_node(self):
if self.root.next is self.root:
return
curnode = self.root.next
while curnode.next is not self.root:
yield curnode
curnode = curnode.next
yield curnode
def __iter__(self):
for node in self.iter_node():
yield node.value
def iter_node_reverse(self):
if self.root.prev is self.root:
return
curnode = self.root.prev
while curnode.prev is not self.root:
yield curnode
curnode = curnode.prev
yield curnode
def test1():
dll = DoubleLinkedList()
dll.append(0)
dll.append(1)
dll.append(2)
dll.append(3)
dll.append(4)
dll.append(5)
assert list(dll) == [0,1,2,3,4,5]
assert [node.value for node in dll.iter_node()] == [0,1,2,3,4,5]
assert [node.value for node in dll.iter_node_reverse()] == [5,4,3,2,1,0]

浙公网安备 33010602011771号