LeetCode155 最小栈
LeetCode 155 最小栈 设计一个支持 push,pop,top 操作,并能在常数时间内检索到最小元素的栈。
push(x) – 将元素 x 推入栈中。 pop() – 删除栈顶的元素。 top() – 获取栈顶元素。 getMin() – 检索栈中的最小元素。
示例:
MinStack minStack = new MinStack(); minStack.push(-2); minStack.push(0); minStack.push(-3); minStack.getMin(); –> 返回 -3. minStack.pop(); minStack.top(); –> 返回 0. minStack.getMin(); –> 返回 -2.
想了一下,这个题目可以用两个栈来实现,一个栈储存正常的数据,另一个栈储存当前的最小值。需要返回最小值的时候直接返回最小栈的栈顶元素即可。弹出栈的时候也是同样的道理。 最后代码如下:
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
class MinStack {
public:
/** initialize your data structure here. */
MinStack() {
}
void push(int x) {
s.push(x);
if (min_s.empty())
{
min_s.push(x);
}
else
{
if (x<min_s.top())
{
min_s.push(x);
}
else
{
min_s.push(min_s.top());
}
}
}
void pop() {
s.pop();
min_s.pop();
}
int top() {
return s.top();
}
int getMin() {
return min_s.top();
}
private:
stack<int> s;
stack<int> min_s;
};
看到网上别人的解法也很神奇,可以不用两个栈就可以实现该算法,可以用一个栈的两个数据位来保存数据,一个数据保存真正 push 进来的值,另一个数据保存当前的最小值,不过实际原理和用两个栈来实现么什么区别。
本文由作者按照 CC BY 4.0 进行授权