栈排序。
对栈进行排序使最小元素位于栈顶。最多只能使用一个其他的临时栈存放数据,但不得将元素复制到别的数据结构(如数组)中。该栈支持如下操作:push、pop、peek和isEmpty。当栈为空时,peek返回-1。
输入:
[“SortedStack”, “push”, “push”, “peek”, “pop”, “peek”]
[[], [1], [2], [], [], []]
输出:
[null,null,null,1,null,2]
思路:辅助栈
利用辅助栈可以保证每次插入新元素的适合s1都是有序的,
比如s1 = {1, 5, 6, 8, 9}插入7
先把9和8插入s2,s2 = {9, 8}
再把7插入s1,s1 = {1, 5, 6, 7}
再把s2中数字插入s1, s2 = {1, 5, 6, 7, 8, 9}
另一种解法,维护两个栈,原栈为降序,辅助栈为升序
比如s1 = {8, 7, 3} s2 = {}
插入5,因为比s1.top大,把3插入s2中,然后 5插入 s1 中
s1 = {8,7,5} s2={3}
这样既能保证 s1 中的元素一定大于 s2 中元素,也可以使得两个栈都是按顺序排列
不必要像第一种解法一样需要在push的时候就把 s2 中元素重新加入到 s1 中去
class SortedStack {
public:
stack<int>s1;//原栈为降序
stack<int>s2;//辅助栈为升序
SortedStack() {
}
void push(int val) {
while(!s2.empty() && s2.top() > val){//辅助栈中存在比val大的值
s1.push(s2.top());
s2.pop();
}
while(!s1.empty() && s1.top() < val){//原栈中有比val小的值
s2.push(s1.top());
s1.pop();
}
s1.push(val);
}
void pop() {
while(!s2.empty()){//清空辅助栈
s1.push(s2.top());
s2.pop();
}
if(!s1.empty()) s1.pop();
}
int peek() {
while(!s2.empty()){//清空辅助栈
s1.push(s2.top());
s2.pop();
}
if(!s1.empty()) return s1.top();
else return -1;
}
bool isEmpty() {
return s1.empty() && s2.empty();
}
};