题目:
定义栈的数据结构,请在该类型中实现一个能够得到栈的最小元素的min函数。在该栈中,调用min、push和pop的时间复杂度都是O(1)。
看到这个问题,第一反应就是每次压入一个新元素进栈时,将栈里的所有元素排序,让最小的元素位于栈顶,这样就可以在O(1)时间得到最小元素了,但是这种思路不能保证最后压入栈的元素能够最先出栈,因此这个数据结构已经不是栈了。
思路:建立一个辅助栈,将每次的最小元素都保存在该栈中。
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里若是条件出错,则停止运行
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();
}