题目描述:
定义栈的数据结构,请在该类型中实现一个能够得到栈中所含最小元素的min函数(时间复杂度应为O(1))。
时间限制:1秒
空间限制:32768K
思路:利用一个辅助栈来存放最小值,栈顶存放的就是当前数据栈中最小值。
具体操作:
1、在向数据栈中 push 数据时,如果当前元素小于辅助栈栈顶元素,就将当前元素一并加到辅助栈中。
2、如果当前元素大于等于辅助栈栈顶元素,那么就把当前栈顶元素再次 push 到辅助栈中。这样想,当前元素大于等于最小值,那么把当前元素 push 到数据栈中后,数据栈中最小值依然是当前元素未 push 前的最小值,所以应该再次将辅助栈栈顶元素 push 到辅助栈中。
class Solution {
public:
//dataSta 数据栈,minSta 最小值栈
stack<int> dataSta,minSta;
void push(int value) {
dataSta.push(value);
if(minSta.empty()) {
minSta.push(value);
}
else {
int top = minSta.top();
value < top ? minSta.push(value) : minSta.push(top);
}
}
void pop() {//pop 时都要 pop 操作
dataSta.pop();
minSta.pop();
}
int top() {
return dataSta.top();
}
int min() {
return minSta.top();
}
};