定义栈的数据结构,请在该类型中实现一个能够得到栈的最小元素的min函数,在该栈中,调用min,push以及pop的时间复杂度都是O(1):
#include <iostream>
#include <stack>
using namespace std;
template<typename T>
class StackWithMin
{
public:
void push(const T& value);
void pop();
const T& min() const;
private:
stack<T> m_data;
stack<T> m_min;
};
//m_data, m_min为两个栈,前者为原始数据,后者为临时的栈
template<typename T> void StackWithMin<T>::push(const T &value)
{
m_data.push(value);
if (m_min.size() == 0 || value < m_min.top())
m_min.push(value);
else
m_min.push(m_min.top());
}
template<typename T> void StackWithMin<T>::pop()
{
assert (m_data.size() > 0 && m_min.size() > 0);
m_data.pop();
m_min.pop();
}
template<typename T> const T& StackWithMin<T>::min() const
{
assert (m_data.size() > 0 && m_min.size() > 0);
return m_min.top();
}