题目
定义栈的数据结构,请在该类型中实现一个能够得到栈最小元素的min函数。
思路
维护两个栈,一个用来存储数据,还有一个用来记录数据栈中的递减数列。
code C++
class Solution {
private:
stack<int> minst;
stack<int> st;
public:
void push(int value) {
st.push(value);
if(minst.empty()){
minst.push(value);
}
else if(minst.top()>=value){
minst.push(value);
}
}
void pop() {
if(minst.top() == st.top()){
minst.pop();
}
st.pop();
}
int top() {
return st.top();
}
int min() {
return minst.top();
}
};