定义栈的数据结构,请在该类型中实现一个能够得到栈中所含最小元素的min函数(时间复杂度应为O(1))。
class Solution {
public:
void push(int value)
{
this->m_data.push(value);
if(this->m_min.size() == 0 || value < m_min.top())
{
m_min.push(value);
}
else
{
m_min.push(m_min.top());
}
}
void pop()
{
if(m_data.size() <= 0 && m_min.size() <= 0) return;
m_data.pop();
m_min.pop();
}
int top()
{
if(m_data.size() <= 0 && m_min.size() <= 0) return -1;
return m_data.top();
}
int min()
{
if(m_data.size() <= 0 && m_min.size() <= 0) return -1;
return m_min.top();
}
private:
stack<int> m_data; //数据栈
stack<int> m_min; //辅助栈
};