leetcode刷题笔记 面试题 03.05. 栈排序

题目描述：

栈排序。 编写程序，对栈进行排序使最小元素位于栈顶。最多只能使用一个其他的临时栈存放数据，但不得将元素复制到别的数据结构（如数组）中。该栈支持如下操作：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()