带最小值操作的栈
描述
笔记
数据
评测
实现一个带有取最小值min方法的栈,min方法将返回当前栈中的最小值。
你实现的栈将支持push,pop 和 min 操作,所有操作要求都在O(1)时间内完成。
注意事项
如果堆栈中没有数字则不能进行min方法的调用
您在真实的面试中是否遇到过这个题?
样例
如下操作:push(1),pop(),push(2),push(3),min(), push(1),min() 返回 1,2,1
class MinStack {
public:
MinStack() {
// do initialization if necessary
}
void push(int number) {
// write your code here
m1.push(number);
if(m2.empty()||number<=m2.top()) //最初m2.empty()写成m1.empty(),结果找了好久的bug...
m2.push(number);
else m2.push(m2.top());
}
int pop() {
// write your code here
if(m1.empty()) return -1;
if(!m1.empty()){
m2.pop();int m=m1.top();m1.pop();
return m;
}
if (m1.empty()) return -1;
int t = m1.top(); m1.pop();
if (!m2.empty() && m2.top() == t) m2.pop();
return t;
}
int min() {
// write your code here
if(!m2.empty())
return m2.top();
return -1;
}
private:
stack<int> m1,m2;
};