您的位置:首页 > 其它

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

2012-07-14 09:35 260 查看
定义栈的数据结构,要求添加一个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;
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: