包含min函数的栈(剑指Offer-30)
题面
定义栈的数据结构,请在该类型中实现一个能够得到栈的最小元素的 min 函数在该栈中,调用 min、push 及 pop 的时间复杂度都是 O(1)。
示例
1MinStack minStack = new MinStack();
2minStack.push(-2);
3minStack.push(0);
4minStack.push(-3);
5minStack.min(); --> 返回 -3.
6minStack.pop();
7minStack.top(); --> 返回 0.
8minStack.min(); --> 返回 -2.
限制
1各函数的调用总次数不超过 20000 次
思路
除了常规的栈,再维护一个单调栈,从栈顶到栈底递增
常规栈常规添加删除元素
单调栈栈顶元素大于等于待加入元素或者单调栈为空,则入栈;当常规栈的出栈元素等于单调栈的栈顶元素,单调栈出栈
代码
1class MinStack {
2 stack<int> s, t;
3public:
4 /** initialize your data structure here. */
5 MinStack() {
6
7 }
8
9 void push(int x) {
10 if(t.empty() || t.top() >= x)
11 t.push(x);
12 s.push(x);
13 }
14
15 void pop() {
16 if(t.top() == s.top())
17 t.pop();
18 s.pop();
19 }
20
21 int top() {
22 return s.top();
23 }
24
25 int min() {
26 return t.top();
27 }
28};
29
30/**
31 * Your MinStack object will be instantiated and called as such:
32 * MinStack* obj = new MinStack();
33 * obj->push(x);
34 * obj->pop();
35 * int param_3 = obj->top();
36 * int param_4 = obj->min();
37 */