题目:定义栈的数据结构,请在该类型中实现一个能够得到栈的最小元素的min函数。在该栈中,调用min、push及pop的时间复杂度都是O(1).
思路:时间复杂度O(1)?刚开始看到觉得不可能实现,对栈进行排序?那还能满足先入后出吗?所以对原始栈进行操作是不可能的了。那么引入一个新的数据结构,来记录栈中的最小值。如果这种数据结构中只保存栈中的最小值,那么栈弹出一个元素呢,新的栈的最小值是多少?很明显这种数据结构不仅记录最小值,还记录之前的次小值、次次小值…这里用到一个辅助栈,每次入栈,除了在原栈上压入数字,还要在辅助栈栈顶压入当前的最小值,由于辅助栈已经记录了在压入栈之前的最小值,因此压入的元素只需和辅助栈栈顶进行比较就可以了。保持辅助栈与原始栈的一致,push的时候,辅助栈的栈顶也总是push一个当前的最小值。出栈的时候,原始栈和辅助栈同时pop,这样辅助栈的栈顶总是记录着原始栈的状态,即最小值。
template <typename T> class StackWithMin
{
private:
stack<T> m_data;
stack<T> m_min;
public:
void push(const T& value);
void pop();
const T& min() const;
};
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();
}