请你设计一个 最小栈 。它提供 push
,pop
,top
操作,并能在常数时间内检索到最小元素的栈。
力扣(LeetCode)官网 - 全球极客挚爱的技术成长平台
实现 MinStack
类:
MinStack()
初始化堆栈对象。void push(int val)
将元素val推入堆栈。void pop()
删除堆栈顶部的元素。int top()
获取堆栈顶部的元素。int getMin()
获取堆栈中的最小元素。
示例 1:
输入: ["MinStack","push","push","push","getMin","pop","top","getMin"] [[],[-2],[2],[-3],[],[],[],[]] 输出: [null,null,null,null,-3,null,2,-2] 解释: MinStack minStack = new MinStack(); minStack.push(-2); minStack.push(2); minStack.push(-3); minStack.getMin(); --> 返回 -3. minStack.pop(); minStack.top(); --> 返回 2. minStack.getMin(); --> 返回 -2. 提示:
-231 <= val <= 231 - 1
pop
、top
和getMin
操作总是在 非空栈 上调用push
、pop
、top
和getMin
最多被调用3 * 104
次
解题思路:
设计两个栈S1:正常入栈。S2(最小栈):从第一个入栈元素开始,较小的值入此栈,栈顶元素是入栈元素中的最小值。
pop:若S1栈顶元素与S2中的栈顶元素相同,则二者同时出栈
push:S1正常入栈,S2栈顶元素对比每个入栈元素,若比前一个入栈元素小,则入此栈。初始为空栈则第一个入栈元素入此栈。
top:正常返回S1.top();
getMin:return S2栈顶元素。
class MinStack { private: stack<int> s1,s2;//s1:正常栈,s2:min栈 public: /** initialize your data structure here. */ MinStack() { } void push(int x) {//小于等于栈底元素的全入s2栈 s1.push(x); if(s2.empty()) { s2.push(x); } else { if(x<=s2.top())//记住是<=,否则入栈0 1 0,最小栈应该进入0 0,否则就错了 { s2.push(x); } } } void pop() {//若s2栈栈顶元素与此时s1pop出的元素一样,push s1:-1 2 -3 s2:-1 -3 pop s1:-1 2;s2:-1 if(!s2.empty()&&s1.top()==s2.top()) { s1.pop(); s2.pop(); } else { s1.pop(); } } int top() { return s1.top(); } int getMin() { return s2.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->getMin(); */