定义栈的数据结构,要求添加一个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;
}