定义栈的数据结构,请在该类型中实现一个能够得到栈中所含最小元素的min函数(时间复杂度应为O(1))。
class Solution {
public:
//与书中所写思路一致:最小栈栈顶即为当前最小值
void push(int value)
{ //若栈为空,则压入栈st,min_st
if(st.empty())
{
st.push(value);
min_st.push(value);
}
else
{
st.push(value);
//若新入栈的元素小于最小栈栈顶元素,则此元素为新的最小元素,入最小栈
if(st.top()<min_st.top())
min_st.push(value);
//若新入栈的元素大于最小栈栈顶元素,则最小元素不变
else
min_st.push(min_st.top());
}
}
//出栈处理一样:栈顶与最小栈同时出栈即可
void pop()
{
if(st.empty())
return;
st.pop();
min_st.pop();
}
//直接取st栈顶元素
int top() {
if(st.empty())
return NULL;
return st.top();
}
int min() {
if(min_st.empty())
return NULL;
return min_st.top();
}
private:
//创建两个栈,st正常存放入栈元素,min_st存放最小元素
stack<int> st;
stack<int> min_st;
};