【在线编程】包含min函数的栈
【问题描述】
定义栈的数据结构,请在该类型中实现一个能够得到栈中所含最小元素的min函数(时间复杂度应为O(1))。
【解题思路 & Java实现】
方法: stack栈用来存储数据,并借用辅助栈min同步记录stack每个状态下的最小值,也就是说让min的栈顶始终保持当前stack栈的最小元素,所以在stack栈进行 push 操作时,比较node和min栈顶元素的大小,向min栈push较小值;在stack进行pop操作时,同时对min进行pop操作。
注意:pop是删除栈的栈顶元素,peek是返回栈的栈顶元素,但不删除。
import java.util.Stack;
public class Solution {
Stack<Integer> stack = new Stack<>();
Stack<Integer> min = new Stack<>();
public void push(int node) {
stack.push(node);
if (min.isEmpty() || node < min.peek()) {
min.push(node);
} else {
min.push(min.peek());
}
}
public void pop() {
if (stack.isEmpty()) {
throw new RuntimeException("Stack is empty!");
}
stack.pop();
min.pop();
}
public int top() {
return stack.peek();
}
public int min() {
return min.peek();
}
}