本题目选自July大神博客系列【微软面试100题】:july大神,该系列我主要用来记录我的学习笔记。
题目描述:设计包含 min 函数的栈
要求:定义栈的数据结构,添加一个min函数,能够得到栈的最小元素,要求函数min、push及pop时间复杂度都是O(1)
分析思路:(本题也见于《剑指offer》的面试题21)
本题要注意, 如果仅仅是靠增加一个成员变量来保存最小元素的值是不行的,因为当pop这个最小元素之后,其下一个最小元素该如何更新?
建立一个辅助栈,每次push的时候,都把当前最小元素压入辅助栈,比如入栈顺序是,3,5,2,4,1.那么辅助栈压入的顺序是3,3,2,2,1。
以下是这个栈的声明和实现:
template<class T>
class StackWithMin<T>
{
public:
//省略构造函数和复制控制
void push(const T& value);
void pop();
const T& min() const;
private:
stack<T> m_data;//使用adaptor,文件头应#include<stack>
stack<T> m_min;//m_data是数据栈,m_min是辅助栈
}
实现
template<class T>
void StackWithMin<T>::push(const T& value)
{
m_data.push(value);
if(m_min.size() == 0 || m_min.top() > value)
m_min.push(value);
else
m_min.push(m_min.top());
}
template<class T>
void StackWithMin<T>::pop()
{
if(m_data.size() > 0 && m_min.size() > 0)
{
m_data.pop();
m_min.pop();
}
}
template<class T>
const T& StackWithMin<T>::min()
{
if(m_data.size() <= 0 && m_min.size() <= 0)
throw exception("stack is empty!");
return m_min.top();
}