题目:
编写程序,对栈进行排序使最小元素位于栈顶。最多只能使用一个其他的临时栈存放数据,但不得将元素复制到别的数据结构(如数组)中。该栈支持如下操作:push
、pop
、peek
和 isEmpty
。当栈为空时,peek
返回 -1。
示例:
输入:
["SortedStack", "push", "push", "peek", "pop", "peek"]
[[], [1], [2], [], [], []]
输出:
[null,null,null,1,null,2]
解题思路:
创建一个辅助栈s2,当要将val入栈s1时,判断s1的元素是否小于val,只要小于,就将元素放入s2栈中,暂时保存起来,直到找到比val大的就停止,然后将val入栈s1,再将刚刚在s2中保存的所有元素依次入栈s1,这样就能始终保持s1中的元素自底向上是按从大到小的顺序存放的。
eg.
当前s1栈的元素:5,3,2,1 ,val=4
1.通过比较,先将1,2,3,依次出栈并存放到s2中,此时 s2: 1,2,3 s1:5
2.将val入栈。 此时:s1:5,4
3.再将元素归还给s1。此时:s1:5,4,3,2,1 s2:null
源代码如下:
class SortedStack {
public:
stack<int> s1;//原始栈
stack<int> s2;//辅助栈,该栈用来保存s1栈中所有比val小的值
SortedStack() {
}
//元素入栈
void push(int val) {
//如果栈为空,直接入栈就可以了,此时该元素就是栈中最小元素
if(s1.empty())
{
s1.push(val);
return ;
}
//否则,要判断s1栈顶元素与val的大小
//如果s1的栈顶元素小于此时要入栈的元素,就要保存这个小元素
while(!s1.empty()&&s1.top()<val)
{
//将s1的栈顶元素放入到辅助栈s2中
s2.push(s1.top());
//栈顶元素出栈
s1.pop();
}
//val入栈
s1.push(val);
//如果此时s2栈不为空,说明刚刚保存过最小的元素,要将这个元素归还给s1
while(!s2.empty())
{
s1.push(s2.top());
s2.pop();
}
}
//元素出栈
void pop() {
//栈为空,无法出栈,直接return
if(s1.empty()) return ;
//否则出栈即可
s1.pop();
}
//返回下一个要访问的元素
int peek() {
//栈为空,直接return
if(s1.empty()) return -1;
//返回栈顶元素
return s1.top();
}
//判断栈是否为空
bool isEmpty() {
return s1.empty();
}
};