面试题 03.05. 栈排序
栈排序。 编写程序,对栈进行排序使最小元素位于栈顶。最多只能使用一个其他的临时栈存放数据,但不得将元素复制到别的数据结构(如数组)中。该栈支持如下操作:push、pop、peek 和 isEmpty。当栈为空时,peek 返回 -1。
定义一个主栈和辅助栈的方法,辅助栈中存储比val小的值。
class SortedStack:
def __init__(self,**args):
#定义一个主stack,和一个辅助的help
self.stack=[]
self.help=[]
def push(self, val: int) -> None:
#当栈不为空并且栈的最后一个元素小于val的时候
#self.help.append(tmp)这一步骤是将stack中小于val的都先存放在help中
while self.stack and self.stack[-1]<val:
tmp = self.stack.pop()
self.help.append(tmp)
#相当于当栈的最后一个元素大于val或者栈为空时将val加入到栈中
self.stack.append(val)
#最后将help中的元素pop出来,加入到栈中,就实现了排序
while self.help:
self.stack.append(self.help.pop())
def pop(self) -> None:
if self.stack:
return self.stack.pop()
else:
return -1
def peek(self) -> int:
if self.stack:
return self.stack[-1]
return -1
def isEmpty(self) -> bool:
if self.stack:
return False
if not self.stack and not self.help:
return True
# 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()