题目:定义栈的数据结构,请在该类型中实现一个能够得到栈最小元素的min函数。
分析:定义两个栈为私有数据,sta保存所有数据,sta_min的top保存最小的那个数据;
入栈:数据压入sta,若sta_min为空,或者数据比sta_min的top小,将该数压入sta_min,现在sta_min的top为sta的最小值;
出栈:若sta和sta_min的top相等,二者同时弹出(此时top值为最小值),若不等,sta弹出。
class Solution {
public:
void push(int value)
{
sta.push(value);
if(sta_min.empty())
sta_min.push(value);
else if(sta_min.top()>value)
sta_min.push(value);
}
void pop()
{
if(sta.top()==sta_min.top())
sta_min.pop();
sta.pop();
}
int top()
{
return sta.top();
}
int min()
{
return sta_min.top();
}
private:
stack<int>sta;
stack<int>sta_min;
};