定义栈的数据结构,请在该类型中实现一个能够得到栈的最小元素的min函数。在该栈中,调用min、push及pop的时间复杂度都是O(1)。
思路:push 和pop肯定都是符合要求的,那么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&StartWithMin<T>::min() const
{
assert(m_data.size()>0&&m_min.size()>0);
return m_min.top();
}