####问题分析
要记录从当前栈顶到栈底元素的最小值,很容易想到用一个变量,每push一个元素更新一次变量的值。那么问题来了,当执行pop操作时,上一次的最小值就找不到了。
####解决方法
方法1、
使用一个栈。元素x入栈时,执行一次push(x),再push(min),min表示当前栈顶到栈底元素最小值;元素出栈时,执行两次pop()
看上图中的例子,蓝色的为最小值,
push:先比较当前栈顶的最小值与插入的值,记录两个中的最小值,先push 插入值,再push记录的最小值。
pop:pop 两次
getmin:top栈顶的值。
方法2、
使用两个栈,一个栈s,保存push的值,另一个栈作为辅助栈,保存当前最小值
push : 栈s插入值push(data),辅助栈min,如果为空或者插入的值data小于等于辅助栈的栈顶元素,辅助栈也插入data。
pop:栈s pop操作,辅助栈中的栈顶元素如果等于栈s的栈顶的话也执行pop操作。
getmin:获取min栈的栈顶元素。
template<class T>
class StackWithMin
{
public:
StackWithMin()
{}
void Push(T data)
{
s.push(data);
if (minStack.empty() || minStack.top() >= data)
minStack.push(data);
}
void Pop()
{
if (s.top() == minStack.top())
minStack.pop();
s.pop();
}
T& MinData()
{
if (!minStack.empty())
return minStack.top();
}
private:
stack<T> s;
stack<T> minStack;
};
void Test3()
{
StackWithMin<int> min_stack;
min_stack.Push(1);
min_stack.Push(2);
min_stack.Push(3);
min_stack.Push(4);
cout << min_stack.MinData() << endl;
}