2015-05-15 18:14:39 +08:00
|
|
|
class Node(object):
|
2015-07-03 11:05:10 +08:00
|
|
|
|
2015-05-15 18:14:39 +08:00
|
|
|
def __init__(self, data):
|
|
|
|
self.data = data
|
|
|
|
self.next = None
|
|
|
|
|
|
|
|
class Queue(object):
|
2015-07-03 11:05:10 +08:00
|
|
|
|
2015-05-15 18:14:39 +08:00
|
|
|
def __init__(self):
|
|
|
|
self.first = None
|
|
|
|
self.last = None
|
|
|
|
|
|
|
|
def enqueue(self, data):
|
|
|
|
node = Node(data)
|
|
|
|
if self.first is None and self.last is None:
|
|
|
|
self.first = node
|
|
|
|
self.last = node
|
|
|
|
else:
|
|
|
|
self.last.next = node
|
|
|
|
self.last = node
|
|
|
|
|
|
|
|
def dequeue(self):
|
|
|
|
# Empty list
|
|
|
|
if self.first is None and self.last is None:
|
|
|
|
return None
|
2015-07-03 11:05:10 +08:00
|
|
|
|
2015-05-15 18:14:39 +08:00
|
|
|
# Remove only element from a one element list
|
|
|
|
elif self.first == self.last:
|
|
|
|
data = self.first.data
|
|
|
|
self.first = None
|
|
|
|
self.last = None
|
|
|
|
return data
|
|
|
|
else:
|
|
|
|
data = self.first.data
|
|
|
|
self.first = self.first.next
|
|
|
|
return data
|