面试题21. 包含min函数的栈
原创
©著作权归作者所有:来自51CTO博客作者mb6304a73bed12a的原创作品,请联系作者获取转载授权,否则将追究法律责任
题目描述
定义栈的数据结构,请在该类型中实现一个能够得到栈最小元素的min函数。
思路:
考察栈的使用。代码中用到两个栈:
a. 用栈 stack 记录所有元素的值
b. 用辅助栈 min_stack 始终记录每次压栈时的最小元素,min_stack的栈顶就是当前的最小元素
具体操作如下:
- 如果A小于等于 min_stack 的栈顶元素,则同时向stack和min_stack压入A
- 否则,向min_stack压入min_stack的栈顶元素,向stack压入A
java 代码如下:
import java.util.Stack;
public class Solution
Stack<Integer> stack = new Stack();
Stack<Integer> min_stack = new Stack();
public void push(int node) {
if(min_stack.isEmpty() || node <= min_stack.peek()) {
min_stack.push(node);
}else {
min_stack.push(min_stack.peek());
}
stack.push(node);
}
public void pop() {
min_stack.pop();
stack.pop();
}
public int top() {
return stack.peek();
}
public int min() {
return
python 代码如下:
# -*- coding:utf-8 -*-
class Solution:
stack = []
min_stack = []
def push(self, node):
if len(self.stack) == 0 or node <= self.stack[-1]:
self.min_stack.append(node)
else:
self.min_stack.append(self.min())
self.stack.append(node)
def pop(self):
self.min_stack.pop()
self.stack.pop()
def top(self):
return self.stack[-1]
def min(self):
return self.min_stack[-1]