包含min函数的栈 【微软面试100题 第二题】

题目要求：定义栈的数据结构，请在该类型中实现一个能够得到栈的最小元素的min函数。在该栈中，调用min、push及pop的时间复杂度都是O(1).

参考题目：剑指offer第21题.

题目分析：

　　1.采用面向对象思想，定义类StackWithMin，包含min、push和pop等方法；

　　2.StackWithMin类中包含两个栈：一个数据栈，一个辅助栈。数据栈中是每次压入的实际数据，辅助栈中是对应数据栈的当前结点的最小值。

　　　　解释：假设数据栈为stackdata,辅助栈为stackmin;令压栈数据依次为5-3-4-1-2；

　　　　　　　则压入5：stackdata:5 , stackmin:5

　　　　　　　　　　压入3：----->stackdata:5-3 , stackmin:5-3

　　　　　　　　　　　　压入4：----->stackdata:5-3-4 , stackmin:5-3-3

　　　　　　　　　　　　　　压入1：----->stackdata:5-3-4-1 , stackmin:5-3-3-1

　　　　　　　　　　　　　　　　　压入2：----->stackdata:5-3-4-1-2, stackmin:5-3-4-1-1

　　　　stackdata即为压入的实际数据，stackmin中每次压入的数据和之前stackmin中的输入比较，如果比之前的小则压入当前，如果比之前的大则压入之前的数据。

#include <iostream>
#include <stack>
#include <cassert>

using namespace std;

template<class T>
class stackWithMin
{
public:
	stackWithMin(){}
	~stackWithMin(){}
	void push(const T& value);
	void pop();
	const T& min() const;
	void printStack();
private:
	stack<T> stackData;
	stack<T> stackMin;
};
template<class T>
void stackWithMin<T>::push(const T& value)
{
	stackData.push(value);

	//辅助栈为空或者压入的元素小于辅助栈的栈顶元素，则压入当前元素；否则压入栈顶元素
	if(stackMin.empty() || stackMin.top()>value)
		stackMin.push(value);
	else
		stackMin.push(stackMin.top());
}
template<class T>
void stackWithMin<T>::pop()
{
	assert(stackData.size()>0 && stackMin.size()>0);

	stackData.pop();
	stackMin.pop();
}
template<class T>
const T& stackWithMin<T>::min() const
{
	assert(stackData.size()>0 && stackMin.size()>0);

	return stackMin.top();
}
template<class T>
void stackWithMin<T>::printStack()
{
	stack<T> tmp;
	
	cout << "当前栈中元素有：";
	while(stackData.size())
	{
		tmp.push(stackData.top());
		stackData.pop();
	}
	
	while(tmp.size())
	{
		stackData.push(tmp.top());
		cout << tmp.top();
		tmp.pop();
	}
	cout << endl;
}
int main(void)
{
	stackWithMin<int> minStack;

	minStack.push(5);
	minStack.printStack();
	cout << "最小元素为：" << minStack.min() << endl;

	minStack.push(3);
	minStack.printStack();
	cout << "最小元素为：" << minStack.min() << endl;

	minStack.push(4);
	minStack.printStack();
	cout << "最小元素为：" << minStack.min() << endl;

	minStack.push(1);
	minStack.printStack();
	cout << "最小元素为：" << minStack.min() << endl;

	minStack.push(2);
	minStack.printStack();
	cout << "最小元素为：" << minStack.min() << endl;

	return 0;
}