这里的要求呢,跟我们平时的stack操作是一样的,那什么地方不同呢?多了一个min方法,并且要求时间复杂度为O(1),那该怎么做呢?O(1)的意思就是说,要用了,直接就能拿到,就好比数组直到下标一样,直接取。
那么如果能够将一个栈的栈顶一直存放的都是最小值呢?那么,我们的要求就实现了,借助c++的封装特性,我编写了以下代码,以便实现功能。
#define _CRT_SECURE_NO_WARNINGS 10
#include<iostream>
#include<cstdlib>
#include<stack>
using namespace std;
template<class T>
class MyStack
{
public:
void Push(const T& x)
{
s1.push(x);
if (s2.empty() || x <= s2.top())
s2.push(x);
}
void Pop()
{
if (!s1.empty())
{
if(s2.top()<s1.top())
s1.pop();
else
{
s1.pop();
s2.pop();
}
}
}
T min()
{
if (!s2.empty())
return s2.top();
}
private:
stack<T> s1;
stack<T> s2;
};
void Test1()
{
MyStack<int> mystack;
mystack.Push(4);
mystack.Push(3);
mystack.Push(2);
mystack.Push(1);
cout<<mystack.min()<<endl;
mystack.Pop();
cout << mystack.min() << endl;
mystack.Pop();
cout << mystack.min() << endl;
mystack.Pop();
cout << mystack.min() << endl;
mystack.Pop();
}
int main()
{
Test1();
system("pause");
return 0;
}
s2中的栈顶,始终保存栈中的最小值,这样就可以实现功能。