Update 0225-implement-stack-using-queues.py

Author: aroch17

python/0225-implement-stack-using-queues.py

@@ -4,18 +4,15 @@ def __init__(self):
 
     def push(self, x: int) -> None:
         self.q.append(x)
+        
+        for _ in range(len(self.q) - 1):
+            self.q.append(self.q.popleft())
 
     def pop(self) -> int:
-        for i in range(len(self.q) - 1):
-            self.push(self.q.popleft())
         return self.q.popleft()
 
     def top(self) -> int:
-        for i in range(len(self.q) - 1):
-            self.push(self.q.popleft())
-        res = self.q[0]
-        self.push(self.q.popleft())
-        return res
+        return self.q[0]
 
     def empty(self) -> bool:
         return len(self.q) == 0