您的位置:首页 > 其它

包含min函数的栈

2015-06-10 20:42 239 查看
题目:定义栈的数据结构 请在该类型中实现一个能够得到栈的最小值的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;
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: