原题来自LeetCode:链接
题目描述
栈排序。 编写程序,对栈进行排序使最小元素位于栈顶。最多只能使用一个其他的临时栈存放数据,但不得将元素复制到别的数据结构(如数组)中。该栈支持如下操作:push、pop、peek 和 isEmpty。当栈为空时,peek 返回 -1。
示例1:
输入:
["SortedStack", "push", "push", "peek", "pop", "peek"]
[[], [1], [2], [], [], []]
输出:
[null,null,null,1,null,2]
示例2:
输入:
["SortedStack", "pop", "pop", "push", "pop", "isEmpty"]
[[], [], [], [1], [], []]
输出:
[null,null,null,null,null,true]
说明:
栈中的元素数目在[0, 5000]范围内。
思路
一开始以为只能用一个栈,后来看了评论发现题目说可以使用一个临时栈,加上存储数据所用的栈,就是可以用两个栈的意思。
可以用两个栈就很简单了,我们始终维护一个有序的栈s1,这个栈自顶向下的元素依次增大,即栈顶最小。
当我们需要push一个值val时,首先判断val是否小于等于s1的栈顶:
- 如果是,那就直接push进s1即可
- 如果不是,那就不断取出s1的元素放入辅助栈s2中,直到s1的栈顶元素大于等于val,然后将val值push进s1,最后将s2里全部的值放回s1里面。
pop时直接pop出栈顶即可,因为我们已经保证了栈顶是最小的。
class SortedStack {
public:
stack<int> s1; //存储数据的栈
stack<int