设计一个支持 push ,pop ,top 操作,并能在常数时间内检索到最小元素的栈。
实现 MinStack 类:
MinStack() 初始化堆栈对象。
void push(int val) 将元素val推入堆栈。
void pop() 删除堆栈顶部的元素。
int top() 获取堆栈顶部的元素。
int getMin() 获取堆栈中的最小元素。
输入:
["MinStack","push","push","push","getMin","pop","top","getMin"]
[[],[-2],[0],[-3],[],[],[],[]]
输出:
[null,null,null,null,-3,null,0,-2]
1、辅助栈法
st minSt
push:-2 -2
push:-2,0 -2
push:-2,0,-3 -2,-3
getmin:-3
pop:-2,0 -2
top:0
getmin:-2
class MinStack {
private:
stack<int> st;
stack<int> minSt;
public:
MinStack() {
}
void push(int val) {
st.push(val);
if (minSt.empty() || val <= minSt.top()) {
minSt.push(val);
}
}
void pop() {
if (st.top() == minSt.top()) {
minSt.pop();
}
st.pop();
}
int top() {
return st.top();
}
int getMin() {
return minSt.top();
}
};
2、不使用辅助栈
class MinStack {
private:
stack<int> st;
int minVal;
public:
MinStack() {
minVal = INT_MAX;
}
void push(int val) {
if (val <= minVal) {
st.push(minVal);
minVal = val;
}
st.push(val);
}
void pop() {
if (st.top() == minVal) {
st.pop();
minVal = st.top();
}
st.pop();
}
int top() {
return st.top();
}
int getMin() {
return minVal;
}
};