题目描述
定义栈的数据结构,请在该类型中实现一个能够得到栈中所含最小元素的min函数(时间复杂度应为O(1))。
注意:保证测试中不会当栈为空的时候,对栈调用pop()或者min()或者top()方法。
思路: 建立两个栈s1和s2。s1用来push进所有的value,而s2则用来push进当前的最小值。
- 当s2为空时,两个栈同时push进value;当s2不为空时,s1则push进该value,而该value与s2的栈顶值比较,若小于s2栈顶值,则value也push进s2,若大于,s2则再一次push其栈顶值。
- 举例: 一开始,value为3时,s2为空,s1和s2同时push进3。s1:3, s2:3。
然后,value为2,s2不为空,且2小于s2的栈顶值3,故s1和s2同时push进2。s1:3,2; s2:3,2。
最后,value为4,s2不为空,且4大于s2的栈顶值2,故s1 push进4,s2 push进2。s1:3,2,4; s2:3,2,2。
因此,可以保证s2能存放最小值,实现min函数。
C++
class Solution {
public:
void push(int value) {
s1.push(value); //s1用来push进所有的value
if (!s2.empty()) //s2不为空时
{
if (value < s2.top()) //该value与s2的栈顶值比较
s2.push(value);
else
s2.push(s2.top());
}
else
s2.push(value);
}
void pop() {
s1.pop();
s2.pop();
}
int top() {
return s1.top(); //s1用来push进所有的value
}
int min() {
return s2.top(); //s2则用来push进当前的最小值
}
private:
stack<int>s1;
stack<int>s2;
};