题目来源:牛客网
编程连接
题目描述
定义栈的数据结构,请在该类型中实现一个能够得到栈最小元素的min函数。
解析:
要在O(1)的时间复杂度内得到最小值。那么只能空间换时间了;
设置一个数据栈,一个辅助栈,辅助栈中存储的是每一步的最小值,当前状态的最小值就是辅助栈的栈顶元素,即
把元素压入数据栈时,比较辅助栈栈顶元素和压入的元素的大小,如果压入的元素比辅助栈顶元素小,那么就把压入的元素压入辅助栈;如果压入的元素比辅助栈顶元素大,那么就把辅助栈顶元素再次压入辅助栈;
当从数据栈中弹出元素时,同时也要从辅助栈中弹出元素。
代码:
class Solution {
public:
stack<int> minStack, st;
void push(int value) {
st.push(value);
minStack.push(minStack.empty()||value<minStack.top()?value:minStack.top());
}
void pop() {
minStack.pop();
st.pop();
}
int top() {
return st.top();
}
int min() {
return minStack.top();
}
};