这道题有点难绷,我以为这道题需要用底层的其他容器来实现,甚至想用vector
来实现,后面看了下灵神的题解发现我们只需要重点解决在常数时间内检索到最小元素的需求即可。其余的函数我们都可以通过stack
自带的函数实现。,那么我们如何实现在O(1)
时间内找到栈中的最小元素呢?我们可以维护一个前缀表,前缀表中维护着当前位置及前面插入的元素中的最小元素,因此我们可以定义一个stack<pair<int, int>>
,第一个元素为插入的元素值,第二个值则为最小前缀,用来维护插入当前元素时整个栈中的最小元素。我们在构造函数中插入一个[0, INT_MAX]
(主要是为了后续第一个元素的插入能正常执行),然后后续插入元素时,第一个元素为要插入的元素val
,第二个元素为min(val, getMin())
,这样,我们就维护了栈中的最小元素,当我们调用getMin()
时,直接返回st.top().second
即可。
class MinStack {
public:stack<pair<int, int>> st; //前面存元素,后面存前缀表,维护当前栈内的最小元素MinStack() {st.push({0, INT_MAX});}void push(int val) {st.push({val, min(val, getMin())});}void pop() {st.pop();}int top() {return st.top().first;}int getMin() {return st.top().second;}
};/*** Your MinStack object will be instantiated and called as such:* MinStack* obj = new MinStack();* obj->push(val);* obj->pop();* int param_3 = obj->top();* int param_4 = obj->getMin();*/