定义栈的数据结构,请在该类型中实现一个能够得到栈中所含最小元素的min函数(时间复杂度应为O(1))。
首先我们来分析一下这个题,题目要求时间复杂度为O(1),那么我们可以用空间换取时间,重新开辟一个栈,让该栈仅保存输入元素的最小值。
注意!!!
题目并没有给提供栈,需要在外部首先定义两个栈。
数据栈 - - - dataStsck ;
最小值栈 - - - minStack ;
import java.util.Stack;
public class Test{
//数据栈
Stack<Integer> dataStsck = new Stack();
//最小值栈
Stack<Integer> minStack = new Stack();
public void push(int node) {
dataStsck.push(node);
if(minStack.isEmpty() ||minStack.peek() > node ){
minStack.push(node);
}
}
public void pop() {
//将数据栈中最小值删除时,同时要删除最小栈中对应的元素。
if(dataStsck.pop() == minStack.peek()){
minStack.pop();
}
}
public int top() {
//该方法仅要求返回栈顶值,故直接返回数据栈顶就好了
return dataStsck.peek();
}
public int min() {
//直接输出最小值栈的栈顶元素就好了。
return minStack.peek();
}
}