lintcode12. 帶最小值操作的棧
最近研究逆向,再加上考試,一直沒刷題。逆向算是基本入門了,過幾天有空寫一篇入門級的逆向分享出來。
實現一個帶有取最小值min方法的棧,min方法將返回當前棧中的最小值。
你實現的棧將支援push,pop 和 min 操作,所有操作要求都在O(1)時間內完成。
注意事項
如果堆疊中沒有數字則不能進行min方法的呼叫
樣例
如下操作:push(1),pop(),push(2),push(3),min(), push(1),min() 返回 1,2,1
思路: 要求在o(1)時間內完成,證明求min時不應該與棧的大小有關。
設計一個輔助棧,每push一個數時,同時將當前最小值壓入棧中,即記錄當前棧狀態時的最小值,有點動態規劃的意思。
程式碼:
class MinStack {
public:
stack<int> a;
stack<int> minstack;
int minnumber;
MinStack() {
minnumber=0x7fffffff;
// stack<int> a; // do intialization if necessary
}
/*
* @param number: An integer
* @return: nothing
*/
void push(int number) {
a.push(number);
if(number<minnumber) minnumber=number;
minstack.push(minnumber);
// write your code here
}
/*
* @return: An integer
*/
int pop() {
int c=a.top();
a.pop();
minstack.pop() ;
if(minstack.size()==0) minnumber=0x7fffffff;
else{minnumber=minstack.top();}
return c;
// write your code here
}
/*
* @return: An integer
*/
int min() {
if(a.size()==0) return NULL;
return minnumber;
// write your code here
}
};