题目描述:
定义栈的数据结构,请在该类型中实现一个功能得到栈中所含最小元素的min函数(时间复杂度应为O(1))。
注意:保证测试中不会当栈为空的时候,对栈调用pop()或者min()或top()方法。
解题思路:
用一个辅助栈来保存最小元素,称为最小元素栈。每次压栈操作时, 如果压栈元素比当前最小元素更小, 就把这个元素压入最小元素栈, 原本的最小元素就成了次小元素. 同理, 弹栈时, 如果弹出的元素和最小元素栈的栈顶元素相等, 就把最小元素的栈顶弹出
class Solution {
public:
stack<int> stk;
stack<int> sMin;
void push(int value) {
stk.push(value);
if(!sMin.empty())
{
if(value < sMin.top())
sMin.push(value);
}
else
sMin.push(value);
}
void pop() {
if(sMin.top() == stk.top())
sMin.pop();
stk.pop();
}
int top() {
return stk.top();
}
int min() {
return sMin.top();
}
};