定义栈的数据结构,请在该类型中实现一个能够得到栈最小元素的min函数。
看到这个 题目,第一想法就是,我要不定义一个变量,指向当前栈中最小的值,每次入栈,就和最小值进行比较,如果比最小值小,则最小值指向该值,否则最小值依旧。
但很快遇到问题,如果在出栈时,最小值出来后,比最小值小的值呢?无法查到了啊!
于是思路出来了,用一个辅助栈,第一次将入栈的值放入,接下来每一次,和栈顶的值进行比较,如果比它小,则将该值放入,否则把栈顶的值再次放入。
此题参考剑指offer书的思路。
下面附代码。
package com.algorithm04;
import java.util.Stack;
public class Algorithm30 {
Stack<Integer> date = new Stack<Integer>();
Stack<Integer> min_date = new Stack<Integer>();
public void push(int node) {
date.push(node);
if(min_date.size()==0||min_date.peek()>node){
min_date.push(node);
}else{
min_date.push(min_date.peek());
}
}
public void pop() {
date.pop();
min_date.pop();
}
public int top() {
return date.peek();
}
public int min() {
return min_date.peek();
}
public static void main(String[] args) {
}
}