Sort Stack:实现一个对栈进行排序的程序,使得最小元素在栈顶,只能使用一个辅助栈。
基本解法类似选择排序,依次从栈中找出最小的元素,然后放入一个新的栈中。这种解法需要使用3
个栈,一个原始栈,一个暂存栈和一个结果栈。
除了冒泡排序,还可以使用插入排序,将原始栈的栈顶元素插入到结果栈中合适的位置,这种算法的时间复杂度为O(N ^ 2)
,空间复杂度为O(N)
。
如果能使用更多的栈,也可以尝试归并排序和快速排序。
class SortedStack {
private:
stack<int> s;
public:
SortedStack() {
}
void push(int val) {
stack<int> sTmp;
while(!s.empty() && s.top() < val){
sTmp.push(s.top());
s.pop();
}
s.push(val);
while(!sTmp.empty()){
s.push(sTmp.top());
sTmp.pop();
}
}
void pop() {
if(!s.empty()){
s.pop();
}
}
int peek() {
if(s.empty()) return -1;
else return s.top();
}
bool isEmpty() {
return s.empty();
}
};
/**
* Your SortedStack object will be instantiated and called as such:
* SortedStack* obj = new SortedStack();
* obj->push(val);
* obj->pop();
* int param_3 = obj->peek();
* bool param_4 = obj->isEmpty();
*/