TheAlgorithms-Python/data_structures/queue/queue_on_pseudo_stack.py

59 lines
1.5 KiB
Python
Raw Normal View History

2018-10-19 20:48:28 +08:00
"""Queue represented by a pseudo stack (represented by a list with pop and append)"""
from typing import Any
2019-10-05 13:14:13 +08:00
class Queue:
2018-10-19 20:48:28 +08:00
def __init__(self):
self.stack = []
self.length = 0
def __str__(self):
2019-10-05 13:14:13 +08:00
printed = "<" + str(self.stack)[1:-1] + ">"
2018-10-19 20:48:28 +08:00
return printed
"""Enqueues {@code item}
@param item
item to enqueue"""
2019-10-05 13:14:13 +08:00
def put(self, item: Any) -> None:
2018-10-19 20:48:28 +08:00
self.stack.append(item)
self.length = self.length + 1
"""Dequeues {@code item}
@requirement: |self.length| > 0
@return dequeued
item that was dequeued"""
2019-10-05 13:14:13 +08:00
def get(self) -> Any:
2018-10-19 20:48:28 +08:00
self.rotate(1)
2019-10-05 13:14:13 +08:00
dequeued = self.stack[self.length - 1]
2018-10-19 20:48:28 +08:00
self.stack = self.stack[:-1]
2019-10-05 13:14:13 +08:00
self.rotate(self.length - 1)
self.length = self.length - 1
2018-10-19 20:48:28 +08:00
return dequeued
"""Rotates the queue {@code rotation} times
@param rotation
number of times to rotate queue"""
2019-10-05 13:14:13 +08:00
def rotate(self, rotation: int) -> None:
2018-10-19 20:48:28 +08:00
for i in range(rotation):
temp = self.stack[0]
self.stack = self.stack[1:]
self.put(temp)
self.length = self.length - 1
"""Reports item at the front of self
@return item at front of self.stack"""
2019-10-05 13:14:13 +08:00
def front(self) -> Any:
2018-10-19 20:48:28 +08:00
front = self.get()
self.put(front)
2019-10-05 13:14:13 +08:00
self.rotate(self.length - 1)
2018-10-19 20:48:28 +08:00
return front
"""Returns the length of this.stack"""
2019-10-05 13:14:13 +08:00
def size(self) -> int:
2018-10-19 20:48:28 +08:00
return self.length