问题描述:
定义栈的数据结构,请在该类型中实现一个能够得到栈的最小元素的min函数。在该栈中,调用min,push及pop的时间复杂度都是O(1).
思路:加入一个辅助栈用来存储最小值集合
(这里要注意题目并没有说栈内的元素类型,因此要尽量通用)
import java.util.Stack;
public class Solution {
Stack s1=new Stack();
Stack min=new Stack();
public void push(int node) {
if(min.empty()){
min.push(node);
}else{
int top=(int)min.peek();
if(node<top){
min.push(node);
}else{
min.push(top);
}
}
s1.push(node);
}
public void pop() {
if(!(s1.empty())){
s1.pop();
min.pop();
}
}
public int top() {
return (int)s1.peek();
}
public int min() {
if(min.empty()){
return 0;
}
return (int)min.peek();
}
}