题目描述:
定义栈的数据结构,请在该类型中实现一个能够得到栈中所含最小元素的min函数(时间复杂度应为O(1))。
解决思路:
最开始的解题思路以为要用基本数据结构实现栈,后来发现用数组的话实在是复杂度太高了。
查了一下网上的思路才发现是用基本的栈去实现复杂的栈。
具体思路是维护两个栈,其中一个是基本的栈,另一个保存最小元素的栈。
注意:一定要保持两个栈的同步,pop的时候都pop,push的时候都push,这样才能保证min栈的首元素是当前最小值。
具体代码如下:
import java.util.Stack;
public class Solution {
Stack<Integer> stack=new Stack<Integer>();//存放每次新来的元素
Stack<Integer> minStack=new Stack<Integer>();//以顶元素最小的方式存放当前最小值
public void push(int node) {
stack.push(node);
if(minStack.isEmpty()||node<minStack.peek()){ //一定要考虑minstack为空的情况
minStack.push(node);
}else{
minStack.push(minStack.peek());
}
}
public void pop() {
stack.pop();
minStack.pop();
}
public int top() {
return stack.peek();
}
public int min() {
return minStack.peek();
}
}