modify Queue to use collections.deque instead of a list

prakhar1989 · Sep 26, 2016 · 2f2cdf3 · 2f2cdf3
1 parent ec4e068
commit 2f2cdf3
Showing 1 changed file with 9 additions and 4 deletions.
diff --git a/lists/queue.py b/lists/queue.py
@@ -1,6 +1,12 @@
+from collections import deque
+
+
 class Queue(object):
+    """Wrapper around collections.deque to provide the api consistent with
+    a Queue"""
+
     def __init__(self):
-        self.items = []
+        self.items = deque()
 
     def __str__(self):
         return ("Queue of size: %d" % len(self.items))
@@ -9,11 +15,10 @@ def isEmpty(self):
         return len(self.items) == 0
 
     def enqueue(self, item):
-        self.items.insert(0, item)
+        self.items.append(item)
 
     def dequeue(self):
-        return self.items.pop()
+        return self.items.popleft()
 
     def size(self):
         return len(self.items)
-