Leetcode 刷题

剑指 Offer 30. 包含min函数的栈

  • 难度:简单

题目

定义栈的数据结构,请在该类型中实现一个能够得到栈的最小元素的 min 函数在该栈中,调用 min、push 及 pop 的时间复杂度都是 O(1)。

相同题目: https://leetcode-cn.com/problems/min-stack/

示例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/problems/bao-han-minhan-shu-de-zhan-lcof/
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

题解

class MinStack {
public:
stack<int> stack1, stack2;

/** initialize your data structure here. */
MinStack() {

}

void push(int x) {
stack1.push(x);
if(stack2.empty() or x <= stack2.top()){
stack2.push(x);
}else{
stack2.push(stack2.top());
}
}

void pop() {
stack1.pop();
stack2.pop();
}

int top() {
return stack1.top();
}

int min() {
return stack2.top();
}
};

思路

  1. 定义两个栈,一个用来存数据,一个用来辅助存每一次数据栈入栈时的所有数据最小值
  2. 入栈判断新入的数据x与前面所有入栈数据的大小关系,如果比之前的都小,那么辅助栈入x,否则入上一次所有数据的最小值。出栈无脑两个栈一起出。