题目:定义栈的数据结构,请在该类型中实现一个能够得到栈的最小元素的min函数。在该栈中,调用min、push、及pop的时间复杂度都是O(1)
思路:可以定义两个栈,一个数据栈dataStack,一个辅助栈minStack用来存储每一次的最小元素。
push函数:输入一个value,将value压入数据栈dataStack中,如果minStack为空或value<minStack.top(),则将value堆入堆栈minStack中,否则将minStack.top()再以此堆入到minStack中;pop函数:数据栈每pop一个数,辅助栈pop一个最小值出来,剩下的数值对应剩下的最小值。top函数:输出数据栈的top值;min函数:输出辅助栈的top值。
class Solution {
public:
stack<int> dataStack,minStack;
void push(int value) {
dataStack.push(value);
if(minStack.size()==0 || value<minStack.top())
minStack.push(value);
else
minStack.push(minStack.top());
}
void pop() {
minStack.pop();
dataStack.pop();
}
int top() {
return dataStack.top();
}
int min() {
return minStack.top();
}
};