帶最小值操作的棧
描述
筆記
數據
評測
實現一個帶有取最小值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;
};