学习笔记25——包含min函数的栈

题目：定义栈的数据结构，请在该类型中实现一个能够得到栈的最小元素的min函数。在该栈中，调用min、push及pop的时间复杂度都是O(1)。
思路：新建一个辅助栈，每次压入一个新元素进数据栈时，都把此时的最小元素压入辅助栈，那么就能保证辅助栈的栈顶一直都是最小元素。当元素从数据栈内被弹出以后，同时弹出辅助栈的栈顶元素，此时辅助栈的新栈顶元素就是下一个最小值。
核心代码如下：

template <typename T>
class StackWithMin{
public:
	StackWithMin() {}
	virtual ~StackWithMin() {}

	T& top();
	const T& top() const;

	void push(const T& value);
	void pop();

	const T& min() const;
private:
	stack<T> m_data;  //数据栈，存放栈的所有元素
	stack<T> 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();
}