题目描述
定义栈的数据结构,请在该类型中实现一个能够得到栈中所含最小元素的min函数(时间复杂度应为O(1))。
解法:
import java.util.Stack;
import java.util.Iterator;
public class Solution {
Stack<Integer> stack = new Stack<Integer>();
public void push(int node) {
stack.push(node);
}
public void pop() {
stack.pop();
}
public int top() {
return stack.peek();
}
public int min() {
int min = Integer.MAX_VALUE;
Iterator<Integer> it = stack.iterator();
while(it.hasNext()) {
min = Math.min(it.next(), min);
}
return min;
}
}
方法总结:
stack.push(int node) 将node压入栈中
stack.pop() 移除并返回栈顶对象
stack.peek() (不移除)并返回栈顶对象
iterator.hasNext() 判断序列中是否还有元素
iterator.next() 返回序列的下一个元素
当首次创建某个容器的迭代器时,其指针指向的是该容器第一个元素上方的空白