题目描述
定义栈的数据结构,请在该类型中实现一个能够得到栈中所含最小元素的min函数(时间复杂度应为O(1))。
注意:保证测试中不会当栈为空的时候,对栈调用pop()或者min()或者top()方法。
解答:
思想:方法一 添加一个最小值的栈(这种方法比较好实现)
方法二:下面这个方法是看题解里面一位同学写的,虽然有点问题但是思想是好的。
public class Q_20 {
Stack<Integer> stack = new Stack<>(); int min; public void push(int node) { if (stack.isEmpty()) { min = node; stack.push(node - min); } else { stack.push(node - min);//栈中存放的node-当前最小值的结果 if (min > node) { min = node; } } } public void pop() { int tmp = stack.pop(); if (tmp < 0) { //出栈时,如果栈顶值大于零,说明上一个最小值和当前一样,不需要变动; //如果小于零,则上一个最小值=当前最小值-栈顶值(因为push的时候栈顶的值就是node-min来的) min = min - tmp; } } public int top() { return stack.peek(); } public int min() { return min; }
}