题目:
实现一个栈,要求实现Push(出栈)、Pop(入栈)、Min(返回最小值的操作)的时间复杂度为O(1)
基本思路:
(1)建立两个栈,一个为普通的容纳数据的栈stackData,一个为stackMin的栈用来返回最小值
(2)压入数据时,stackData直接压入数据newNum
` stackMin若为空,直接压入数据newNum。若不为空
若不为空,则则比较要压入数据newNum与stackMin的top()的值,如果numNew<=sstackMin.top()则压入newNum,否则压入sstackMin.top()
(3)弹出数据时,两个stack同时弹出数据
(4)返回最小值操作时,直接返回stackMin的top()
代码如下:
函数声明如下:
template<class T>//模版没写一个函数都要去测试一下
class StackWithMin
{
public:
StackWithMin(){};//将会自动去掉对应的构造函数和析构函数
virtual ~StackWithMin(){};
void Push(const T& x);
void Pop();
const T& Min()const;//不是栈空间的变量,所以可以直接加引用返回,但是不能改变其的值,所以加const
T& Top();
bool Empty() const;
size_t Size()const;
private:
stack<T> StackData;//数据栈,存放栈的所有元素 //使用类模板需要模板参数列表,没有特例话的模版参数
stack<T> StackMin;//辅助栈,存放栈中的最小元素
};
函数定义如下:
template<class T>
void StackWithMin<T>::Push(const T& x)
{
StackData.push(x);
if (StackMin.size() == 0 || x < StackMin.top())
StackMin.push(x);
else
StackMin.push(StackMin.top());
}
template<class T>
void StackWithMin<T>::Pop()
{
assert(StackData.size() > 0 && StackMin.size() > 0);//系统的stack不知道是用链表实现的还是用数组实现的,
//所以这么写,不知道是和0比较还是和NULL比较
StackData.pop();
StackMin.pop();
}
template<class T>
const T& StackWithMin<T>::Min()const
{
assert(StackData.size() > 0 && StackMin.size() > 0);
return StackMin.top();
}
template<class T>
T& StackWithMin<T>::Top()
{
return StackData.top();
}
template<class T>
bool StackWithMin<T>::Empty()const
{
return StackData.size() == 0;
}