定义栈的数据结构,要求添加一个min函数,能够得到栈的最小元素。要求函数min、pop、push的时间复杂度为O(1);
方法:采用一个辅助栈,该栈用来存放最小元素。当进行push操作时,将栈中的最小元素同时push进辅助栈中,当pop操作时同时pop掉辅助栈中的栈顶元素。这样元素栈对应位置的辅助栈中存放的永远是最小元素。
#include
#include
using namespace std;
template
class StackWithMin
{
public:
void push(const T value);
void pop();
T min();
private:
stack dataStack;
stack minStack;
};
template
void StackWithMin::push(T value)
{
dataStack.push(value);
if(minStack.size()==0)
{
minStack.push(value);
}
else if(value
{
minStack.push(value);
}
else
{
minStack.push(minStack.top());
}
}
template
void StackWithMin::pop()
{
dataStack.pop();
minStack.pop();
}
template
T StackWithMin::min()
{
T p;
p=minStack.top();
return p;
}
int main()
{
int t;
StackWithMin s;
s.push(3);
s.push(1);
s.push(5);
s.push(6);
s.push(0);
t=s.min();
cout<
s.pop();
t=s.min();
cout<
return 0;
}