实现一个栈,要求实现Push(出栈)、Pop(入栈)、Min(返回最小值的操作)的时间复杂度为O(1)
在栈中操作的话,push和pop的时间复杂度就是O(1),所以我们只用实现Min(返回最小值的操作)的时间复杂度为O(1),
思想就是用两个栈,一个就是普通的存取数据的栈,另一个为当前未知的最小值,插入数据和删除数据两个栈都进行操作,返回最小值的话,直接对第二个栈操作。
代码如下:
- #include <iostream>
- using namespace std;
- #include <stack>
- template <class T>
- class retmin
- {
- public :
- void pushmin(const T& x)
- {
- _num.push (x);
- if(!_min.empty ())
- {
- T tmp = _min.top ();
- if( x < tmp)
- {
- _min.push (x);
- }
- else
- _min.push (tmp);
- }
- else
- _min.push (x);
- }
- void popmin()
- {
- _num.pop ();
- _min.pop ();
- }
- T Retmin()
- {
- return _min.top ();
- }
- private :
- stack<T> _num;
- stack<T> _min;
- };
- void Test9()
- {
- retmin< int> r1;
- r1.pushmin (5);
- r1.pushmin (4);
- r1.pushmin (3);
- r1.pushmin (6);
- r1.pushmin (0);
- cout<<r1.Retmin ()<<endl;
- }
- int main()
- {
- Test9();
- return 0;
- }