#ifndef TG_STACK_MIN_H
#define TG_STACK_MIN_H
#include<stack>
#include<stdlib.h>
// 面试题30:包含min函数的栈
// 题目:定义栈的数据结构,请在该类型中实现一个能够得到栈的最小元素的min
// 函数。在该栈中,调用min、push及pop的时间复杂度都是O(1)。
template <typename T>
class tg_stack_min
{
public:
tg_stack_min() = default;
~tg_stack_min() = default;
T& top();
const T& top() const;
void push(const T& value);
void pop();
const T& min() const;
bool empty() const;
size_t size() const;
private:
std::stack<T> _data; // 数据栈,存放栈的所有元素
std::stack<T> _min; // 辅助栈,存放栈的最小元素
};
template <typename T>
void tg_stack_min<T>::push(const T& value)
{
_data.push(value);
if(_min.empty() || value < _min.top())
{
_min.push(value);
}
else
{
_min.push(_min.top());
}
}
template <typename T>
void tg_stack_min<T>::pop()
{
assert(_data.size() > 0 && _min.size() > 0);
_data.pop();
_min.pop();
}
template <typename T>
const T& tg_stack_min<T>::min() const
{
assert(_data.size() > 0 && _min.size() > 0);
return _min.top();
}
template <typename T>
T& tg_stack_min<T>::top()
{
assert(_data.size() > 0 && _min.size() > 0);
return _data.top();
}
template <typename T>
const T& tg_stack_min<T>::top() const
{
assert(_data.size() > 0 && _min.size() > 0);
return _data.top();
}
template <typename T>
bool tg_stack_min<T>::empty() const
{
return _data.empty();
}
template <typename T>
size_t tg_stack_min<T>::size() const
{
return _data.size();
}
#endif // TG_STACK_MIN_H