题目
定义栈的数据结构,请在该类型中实现一个能够得到栈最小元素的min函数。
思路
用一个栈data保存数据,用另外一个栈min保存入栈最小的数,在压入时比较当前元素是否小于min栈顶的元素,小于则min压入当前元素,否则再次压入min栈顶的元素,即始终保持min栈顶是当前最小值,同样在data弹出的时候min也要弹出。
代码
package com.zhumq.leetcode;
import java.util.Stack;
public class MinStack {
//存储数据
Stack<Integer> date = new Stack<>();
//存储最小值
Stack<Integer> min = new Stack<>();
public void push(int node) {
date.push(node);
if (min.empty()) {
min.push(node);
} else {
//min栈顶始终是最小值
min.push((min.peek()) > node ? node : min.peek());
}
}
public void pop() {
date.pop();
min.pop();
}
public int top() {
return date.peek();
}
public int min() {
return min.peek();
}
}