要求实现以下功能:
(1)入栈操作push
(2)出栈操作pop
(3)栈大小操作size
(4)栈中最小元素min
主要考虑的是怎样记录栈的最小元素,尤其在某些元素出栈以后,由于不能遍历栈中的元素,因此可以利用另外一个辅助的栈来存放栈中现有所有元素的最小值,即在每次入栈的元素和当前辅助栈的栈顶元素比较,如果入栈元素小与等于辅助栈栈顶元素,则将入栈元素也压入辅助栈,如果入栈元素大,则将辅助栈的栈顶元素再次压入辅助栈。每次出栈时辅助栈也跟着出栈。
#include <iostream>
#include <list>
using namespace std;
class Stack
{
private:
list<int> stack; //链式栈
list<int> min_stack; //存放最小值的链式栈
public:
void push(int value);
void pop();
unsigned int size();
int min();
};
void Stack::push(int value)
{
stack.push_front(value);
if(min_stack.size() == 0)
{
min_stack.push_front(value);
}
else
{
int min = min_stack.front();
if( value < min )
{
min_stack.push_front(value);
}
else
{
min_stack.push_front(min);
}
}
}
void Stack::pop()
{
stack.pop_front();
min_stack.pop_front();
}
unsigned int Stack::size()
{
return stack.size();
}
int Stack::min()
{
return min_stack.front();
}