2.设计包含min 函数的栈。

定义栈的数据结构,要求添加一个min 函数,能够得到栈的最小元素。要求函数min、push 以及pop 的时间复杂度都是O(1)。本题的要求是实现O(1)的算法,空间和时间的关系既使用空间来节约时间;定义2个栈,一个是普通的栈,另一个是保存当前最小的元素的栈,由于每次压站或者出站的时候都进行了比较,所以每次min 的时候只要直接输出;时间复杂度都是0(1);

#include <iostream>
#include <stack>

using namespace std;

bool less(int a,int b)
{
	return ((a < b) ? true : false);
}

template <typename A>
class Stack
{
	friend bool less(A a,A b);
private:
	stack<A> ps;
	stack<A> mins;
public:
	A min()
	{
		return mins.top();
	}

	void push(A a)
	{	
		if ( ps.empty() )
		{
			ps.push(a);
			mins.push(a);
		}
		else
		{
			ps.push(a);
			if ( less(a,mins.top()) )
			{
				mins.push(a);
			}
			else
				mins.push(mins.top());
		}
	}

	void pop()
	{
		mins.pop();
		ps.pop();
	}
};

int main()
{
	Stack<int> s;
	s.push(5);s.push(7);s.push(9);s.push(3);s.push(12);

	cout << s.min() << endl;
	s.pop();

	cout << s.min() << endl;
	s.pop();

	cout << s.min() << endl;
	s.pop();

	cout << s.min() << endl;
	s.pop();

	cout << s.min() << endl;
	s.pop();
	return 0;
}

点赞