Fix LinkedListQueue enqueue and dequeue

RoWisdom · Sep 24, 2017 · ac0b726 · ac0b726
1 parent 2910351
commit ac0b726
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/queue/queue.py b/queue/queue.py
@@ -88,8 +88,9 @@ def enqueue(self, value):
         if not self.front:
             self.front = node
             self.rear = node
-        node.next = self.rear
-        self.rear = node
+        else:
+            self.rear.next = node
+            self.rear = node
         self.top += 1
 
     def dequeue(self):
@@ -98,7 +99,9 @@ def dequeue(self):
         value = self.front.value
         if self.front is self.rear:
             self.front = None
-        self.front = self.front.next
+            self.rear = None
+        else:
+            self.front = self.front.next
         self.top -= 1
         return value