面试题 03.05. 栈排序

题目

栈排序。 编写程序，对栈进行排序使最小元素位于栈顶。最多只能使用一个其他的临时栈存放数据，但不得将元素复制到别的数据结构（如数组）中。该栈支持如下操作：push、pop、peek 和 isEmpty。当栈为空时，peek 返回 -1。

示例1：

输入：
["SortedStack", "push", "push", "peek", "pop", "peek"]
[[], [1], [2], [], [], []]
 输出：
[null,null,null,1,null,2]

分析： 每次插入的新元素都与栈中元素进行比较，将栈中小的元素放入在辅助栈中，然后插入新元素，最后在将辅助栈中的元素放入栈中，保证栈中永远是栈顶最小元素的顺序排序

class SortedStack:

    def __init__(self):
        self.stack = []
        self.stack_tmp = []


    def push(self, val: int) -> None:
        if not self.stack:
            self.stack.append(val)
            return
        while self.stack:
            element = self.peek()
            if val > element:
                self.stack_tmp.append(self.pop())
            else:
                break
        self.stack.append(val)
        while self.stack_tmp:
            self.stack.append(self.stack_tmp.pop())

    def pop(self) -> None:
        if self.stack:
            element = self.stack.pop()
            return element
        return None

    def peek(self) -> int:
        if self.stack:
            return self.stack[-1]
        return -1

    def isEmpty(self) -> bool:
        if self.stack:
            return False
        return True