Polish queue list solution

This commit is contained in:
Donne Martin
2016-02-21 16:21:05 -05:00
parent 76da9ec98d
commit 7222f71f59
2 changed files with 11 additions and 17 deletions

View File

@@ -24,14 +24,11 @@ class Queue(object):
# Empty list
if self.first is None and self.last is None:
return None
data = self.first.data
# Remove only element from a one element list
elif self.first == self.last:
data = self.first.data
if self.first == self.last:
self.first = None
self.last = None
return data
else:
data = self.first.data
self.first = self.first.next
return data
return data