题目:定义栈的数据结构,请在该类型中实现一个能够得到栈的最小元素的 min 函数在该栈中,调用 min、push 及 pop 的时间复杂度都是 O(1)。
示例输入:
示例:
MinStack minStack = new MinStack();
minStack.push(-2);
minStack.push(0);
minStack.push(-3);
minStack.min(); --> 返回 -3.
minStack.pop();
minStack.top(); --> 返回 0.
minStack.min(); --> 返回 -2.
提示:
- 各函数的调用总次数不超过 20000 次
题解:
题目中讲到要求时间复杂度为O(1),因此需要在操作时记录下最小项,具体要求如下:
- 当压入栈时,比较当前最小值与新数据,并更新最小值
- 当pop出栈时,也应当更新当前的最小值,这就包括需要考虑到pop出的数据是否就是最小值,如果就是最小值要怎么找剩下的数据中的最小值,如果通过循环遍历那么时间复杂度必定会超。
考虑到上述要求,我们可以在压入栈的时候压入当前最小值同时压入当前数据,也就是说,压入的一个数据是新数据,一个数据是这个新数据进栈之前,栈内的最小值。在入栈的同时,要注意数据的更新。这样在pop出栈的时候,可以及时地更新当前数据出栈后栈内的最小值,同时需要pop出两个数据。
代码:
class MinStack {
private:
stack<int> my_stack;
int min_num=INT_MAX;
public:
/** initialize your data structure here. */
MinStack() {
}
void push(int x) {
this->my_stack.push(min_num);//当前最小值入栈
if(x<this->min_num)//更新数据
{
this->min_num=x;
}
this->my_stack.push(x);
}
void pop() {
this->my_stack.pop();
this->min_num=this->my_stack.top();//更新当前最小值
this->my_stack.pop();
}
int top() {
return this->my_stack.top();
}
int min() {
return this->min_num;
}
};
/**
* Your MinStack object will be instantiated and called as such:
* MinStack* obj = new MinStack();
* obj->push(x);
* obj->pop();
* int param_3 = obj->top();
* int param_4 = obj->min();
*/