定义栈的数据结构,请在该类型中实现一个能够得到栈的最小元素的 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 次
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/probl...
方法:栈
一个栈用于存放全部元素,另一个栈用于存放不断更新最小值
pop出栈是如果检测到两栈top相等,即同时出栈
min即返回最小栈的top
class MinStack { public: /** initialize your data structure here. */ stack<int>a,b; //b栈用于存放不断更新的最小值 MinStack() { } void push(int x) { a.push(x); if (b.empty() || x <= b.top()) { b.push(x); } } void pop() { if (a.top() == b.top()) { b.pop(); } a.pop(); } int top() { return a.top(); } int min() { return b.top(); } }; /** * 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(); */