剑指Offer之 - 包含min函数的栈

题目：

定义栈的数据结构，请在该类型中实现一个能够得到栈的最小元素的min函数。
在该栈中，调用min,push,pop函数的世界复杂度都是O(1)。

思路：

维护一个辅助栈s2，用来存储当前栈中的最小元素
比如栈s1 :<2,4,3,1,5> ,辅助栈s2就是: <2,2,2,1,1>

代码：

#include<iostream>
#include <stack>
using namespace std;

//功能：定义栈的数据结构，请在该类型中实现一个能够得到栈的最小元素的min函数。
//在该栈中，调用min,push,pop函数的世界复杂度都是O(1)。

//思路：维护一个辅助栈s2，用来存储当前栈中的最小元素
//比如栈s1 :<2,4,3,1,5> ,辅助栈s2就是: <2,2,2,1,1>
struct StackWithMin
{
	stack<int> s1;
	stack<int> s2;
	void push1(int val)
	{
		s1.push(val);
		if(s2.empty() || val <= s2.top())
			s2.push(val);
		else
		{
			s2.push(s2.top());
		}
	}
	int min()
	{
		if(!s2.empty())
			return s2.top();
		else
			return -1;
	}
	void pop1()
	{
		if(!s1.empty() && !s2.empty())
		{
			s1.pop();
			s2.pop();
		}
	}
};

int main()
{
	StackWithMin obj;
	obj.push1(4);
	obj.push1(2);
	obj.push1(3);
	obj.push1(1);
	cout<<obj.min()<<endl;
	obj.pop1();
	cout<<obj.min()<<endl;
}