常数时间求栈的最小/大值
问题描述:
一个栈stack,具有push和pop操作,其时间复杂度皆为O(1)。
设计算法max操作,求栈中的最大值,该操作的时间复杂度也要求为O(1)。
可以修改栈的存储方式,push,pop的操作,但是要保证O(1)的时间复杂度,空间时间复杂度无要求。
算法描述:
定义一个模板类,一个存储所有元素的栈,一个存储所有最小值的栈 _minIndex。
1. 当push入栈的元素小于当前最小元素,将该元素压入最大值栈Sm;
2. Sm栈顶始终保存栈中当前的最小元素;
3. 当前最大元素被pop出栈时,将Sm栈顶的对应最小元素也弹出栈。
min操作即为获得Sm栈顶最小元素。
源码如下:
#include <iostream>
#include <vector>
using namespace std;
template<typename T>class CStack
{
public:
CStack();
~CStack();
void pop(T value);
void push(T value);
const T getmin() const;
private:
vector<T> _data;
vector<size_t> _minIndex;
};
template<typename T>CStack<T>::CStack()
{
}
template<typename T>CStack<T>::~CStack()
{
}
template<typename T>void CStack<T>::pop(T value)
{
if(_data.empty())
return;
else
{
value=_data.back();
_data.pop_back();
if(value==_minIndex.back())
{
_minIndex.pop_back();
}
}
};
template<typename T>void CStack<T>::push(T value)
{
_data.push_back(value);
if(_minIndex.empty()||value<_minIndex.back())
_minIndex.push_back(value);
}
template<typename T>const T CStack<T>::getmin() const
{
if(_data.empty()||_minIndex.empty())
return -1;
return _minIndex.back();
}
int main()
{
CStack<int> sa;
sa.push(5);
sa.push(4);
sa.push(8);
sa.push(4);
sa.push(19);
int a;
a=sa.getmin();
cout<<a<<endl;
return 0;
}