题目:定义栈的数据结构 请在该类型中实现一个能够得到栈的最小值的min函数 在该栈中 调用min、push及pop的时间复杂度都是O(1)
#include <iostream>
#include <stack>
#include <assert.h>
using namespace std;
template <class T>
struct stackmin
{
stack<T> s1;
stack<T> s2;
void push(T a);
void pop();
const T &min() const;
};
template<class T>void stackmin<T>::push(T a)
{
s1.push(a);
if(s2.size()<=0)
s2.push(a);
else
{
int t=s2.top();
if(t<a)
s2.push(t);
else
s2.push(a);
}
}
template<class T>void stackmin<T>::pop()
{
assert(s1.size() >0 && s2.size()>0);
s1.pop();
s2.pop();
}
template<class T>const T &stackmin<T>::min()const
{
return s2.top();
}
int main()
{
stackmin<int> st;
st.push(1);
st.push(3);
st.push(2);
st.push(1);
st.push(3);
st.push(2);
cout<<st.min()<<endl;
st.pop();
cout<<st.min()<<endl;
}