稍微考虑了一下,只用list (stack 其实更简单,且符合题意,但都是线性结构就都可以) 看能不能搞定,但是发现没有简单办法去track min,比如pop出去的正好是min,就不好处理了。。。
所以加了个PriorityQueue来辅助,问题很容易解决了。
注意的是:.remove() 的方法,在ArrayList和PQ中有不同的输入,PQ的输入只能是 object,不能表示index,但是ArrayList中是可以index也可以object,如果是ArrayList<Integer>,那么删除的时候,list.remove(2) 删除的是index=2的元素,list.remove(new Integer(2)) 才是删除元素2.
这个问题在此链接中有解释:http://stackoverflow.com/questions/4534146/properly-removing-an-integer-from-a-listinteger
public class MinStack {
/** initialize your data structure here. */
List<Integer> list;
PriorityQueue<Integer> queue;
public MinStack() {
list= new ArrayList<Integer>();
queue= new PriorityQueue<Integer>();
}
public void push(int x) {
list.add(x);
queue.add(x);
}
public void pop() {
//queue.poll();
queue.remove(list.get(list.size()-1));
list.remove(list.size()-1);
}
public int top() {
return list.get(list.size()-1);
}
public int getMin() {
return queue.peek();
}
}
/**
* Your MinStack object will be instantiated and called as such:
* MinStack obj = new MinStack();
* obj.push(x);
* obj.pop();
* int param_3 = obj.top();
* int param_4 = obj.getMin();
*/