包含min函数的栈
定义栈的数据结构,请在该类型中实现一个能够得到栈中所含最小元素的min函数(时间复杂度应为O(1))。
import java.util.Stack;
import java.util.Iterator;
public class StackMin {
public static void main(String[] args) {
// TODO Auto-generated method stub
Solution solution = new Solution();
solution.push(1);
solution.push(2);
solution.push(3);
solution.push(4);
int min = solution.min();
System.out.println(min);
}
}
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();//peek()函数返回栈顶的元素,但不弹出该栈顶元素。
}
/*
关于Iterator主要有三个方法:hasNext()、next()、remove()
hasNext:没有指针下移操作,只是判断是否存在下一个元素
next:指针下移,返回该指针所指向的元素
remove:删除当前指针所指向的元素,一般和next方法一起用,这时候的作用就是删除next方法返回的元素
*/
public int min() {
int min = stack.peek();
int tmp = 0;
Iterator<Integer> iterator = stack.iterator();
while (iterator.hasNext()) {
tmp = iterator.next();
if (min > tmp) {
min = tmp;
}
}
return min;
}
}