定义栈的数据结构,请在该类型中实现一个能够得到栈最小元素的min函数。
思路分析:
设置两个栈,一个用于存储数据dataStack,另一个用于存储最小值的栈minStack。入栈的时候,将node值与minStack中的peek值进行比较,若node较小,则将node压入minStack;否则,将minStack中的peek值压入minStack。
import java.util.Stack;
public class Solution {
Stack<Integer> dataStack=new Stack<>();
Stack<Integer> minStack=new Stack<>();
public void push(int node) {
dataStack.push(node);
if(minStack.empty()||node<minStack.peek())
{
minStack.push(node);
}
else
{
minStack.push(minStack.peek());
}
}
public void pop() {
dataStack.pop();
minStack.pop();
}
public int top() {
return dataStack.peek();
}
public int min() {
return minStack.peek();
}
}