题目
定义栈的数据结构,请在该类型中实现一个能够得到栈的最小元素的 min 函数在该栈中,调用 min、push 及 pop 的时间复杂度都是 O(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.
思路
思路一:用栈实现:
- 先建立两个栈A、B,A正常使用,B用来存储最小值;
- 在往A中插入元素的同时,判断B是否为空或者B的栈顶元素是否小于这个新插入的元素,如果B为空或者B的栈顶元素大于新插入的元素,就将这个元素同时插入的B中;
- 在删除A的栈顶元素时,判断B的栈顶元素是否等于这个要删除的元素,如果等于,那么B的这个栈顶元素也要删除。因为A 的这个元素删除了,之后所进行的找最小值的操作,都是在删除这个栈顶元素后的栈中寻找的,不包括要删除的这个栈顶元素,所以B的这个和A相同的栈顶元素也要删除。
- 经过上面的操作,B的栈顶元素就是最小值,所以找最小值直接取B的栈顶元素即可
思路二:用数组实现:
- 用数组元素实现相对简单。先建立一个较大的数组(也可在下面进行扩容操作),数组的长度为0。
- 每插入一个元素,就是将该元素赋给数组的当前长度,并且数组的长度+1。
- 每 删除一个元素,就是使数组的长度-1。
- 根据先进后出,取栈顶元素,就相当于取数组的最后一个元素。
- 取最小值则是将数组遍历比较即可。
代码
法一:用栈实现
class MinStack {
Stack<Integer> A = new Stack<>();
Stack<Integer> B = new Stack<>();//记录最小元素的栈
/** initialize your data structure here. */
public MinStack() {
}
public void push(int x) {
//判断插入的元素是否为最小的值
A.add(x);
//此处一定得是大于等于,因为他可能会插入多个相同的值,如果只按一个算,会造成溢出
if(B.isEmpty() || B.peek() >= x){
B.add(x);
}
}
public void pop() {
//判断栈顶元素是不是最小的
if(B.peek().equals(A.pop())){
B.pop();
}
}
public int top() {
return A.peek();
}
public int min() {
return B.peek();
}
}
法二:用数组实现
class MinStack {
private int[] data = new int[10000];
int size = 0;
/** initialize your data structure here. */
public MinStack() {
}
public void push(int x) {
data[size] = x;
size++;
}
public void pop() {
if(size == 0){
return;
}
size--;
}
public Integer top() {
if(size == 0){
return null;
}
return data[size - 1];
}
public Integer min() {
if(size == 0){
return null;
}
int ret = data[0];
for(int i = 1;i < size;i++){
if(data[i] < ret){
ret = data[i];
}
}
return ret;
}
}
来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/bao-han-minhan-shu-de-zhan-lcof
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。