题目描述:
栈排序。 编写程序,对栈进行排序使最小元素位于栈顶。最多只能使用一个其他的临时栈存放数据,但不得将元素复制到别的数据结构(如数组)中。该栈支持如下操作:push、pop、peek 和 isEmpty。当栈为空时,peek 返回 -1。
我的思路:建立一个辅助栈,使得每次push的元素能够插入到它该去的位置,有点类似于插入排序,整个栈是有序的。
class SortedStack:
def __init__(self):
self.data = []
self.helper = []
def push(self, val: int) -> None:
if not self.data:
self.data.append(val)
else:
if self.data[-1] < val:#如果插入的元素比栈顶元素大,那就把栈中比val小的元素转移到辅助栈中,插入val后再转移回来
while self.data and self.data[-1] < val:
self.helper.append(self.data.pop())
self.data.append(val)
while self.helper:
self.data.append(self.helper.pop())
else:
self.data.append(val)
def pop(self) -> None:
if not self.data:
return -1
return self.data.pop()
def peek(self) -> int:
if not self.data:
return -1
return self.data[-1]
def isEmpty(self) -> bool:
return not self.data
# Your SortedStack object will be instantiated and called as such:
# obj = SortedStack()
# obj.push(val)
# obj.pop()
# param_3 = obj.peek()
# param_4 = obj.isEmpty()