题目及测试
package sword021;
/* 题目描述:定义栈的数据结构,请在该类型中实现一个能够得到栈的最小元素的min函数。在该栈中,调用min、push以及pop的时间复杂度为O(1)
*/
public class main {
public static void main(String[] args) {
test();
}
private static void test() {
Solution solution = new Solution();
int rtn;
long begin = System.currentTimeMillis();
solution.push(1);
solution.push(2);
solution.push(0);
long end = System.currentTimeMillis();
System.out.print(solution.min()+" ");
System.out.println();
System.out.print(solution.top()+" ");
System.out.println();
solution.pop();
System.out.print(solution.min()+" ");
System.out.println();
System.out.print(solution.top()+" ");
System.out.println();
System.out.println("耗时:" + (end - begin) + "ms");
System.out.println("-------------------");
}
}
解法1(成功)
首先这个栈要具有普通栈所具有的push()和pop()方法,那么内部一定包含一个Stack。至于还要能实现min函数,而且还是在O(1)时间复杂度内,我们不得不考虑用额外的空间。
如果直接使用一个int变量存储当前的最小值,我们的确可以获得最小值,但是当栈pop()了以后,我们无法获得次小值。我们需要一个数据结构来动态保存每个时刻的最小值,每当push()和pop()的时候,同样更新这个数据结构,而且时间复杂度也是O(1),那么看来需要额外O(n)的空间了。可以选择栈或者一个链表(实质上当作栈使用)。我们使用一个辅助栈来实现,该栈和主栈的大小相同,栈顶存放的是当前栈内的最小值。
package sword021;
import java.util.Stack;
public class Solution {
//数据栈
private Stack<Integer> stackData;
//辅助栈
private Stack<Integer> stackMin;
public Solution(){
stackData=new Stack<Integer>();
stackMin=new Stack<Integer>();
}
//入栈操作
public void push(int node) {
stackData.push(node);
if(stackMin.isEmpty() || node < min()) {
stackMin.push(node);
}else {
stackMin.push(min());
}
}
//出栈操作
public void pop() {
stackData.pop();
stackMin.pop();
}
//返回栈顶的元素
public int top() {
return stackData.peek();
}
//得到栈的最小元素
public int min() {
return stackMin.peek();
}
}