每天一道算法题——包含min函数的栈

来源:互联网 发布:越狱重启后软件消失 编辑:程序博客网 时间:2024/06/10 02:15

前言
这段时间到了公司,各种事情有点多。好几天没有坚持做算法题了。今天重新捡起来。每天半个小时,剑指offer!
题目描述
定义栈的数据结构,请在该类型中实现一个能够得到栈最小元素的min函数。
测试用例:
[“PSH3”,”MIN”,”PSH4”,”MIN”,”PSH2”,”MIN”,”PSH3”,”MIN”,”POP”,”MIN”,”POP”,”MIN”,”POP”,”MIN”,”PSH0”,”MIN”]
对应输出应该为:
3,3,2,2,2,3,3,0

分析:
这个问题的重点在于怎么去记住栈中最小的那个值,而且在删去这个最小值的时候,最小值能够得到更新。这里采用创建一个辅助栈的方法帮助记忆栈中的最小值。当获得最小值时,置于栈顶,在push和最小值等值的值时,也加入辅助栈。在pop时和辅助栈顶的值进行比较,如果相等就也进行pop操作,否则就保持原样。调用min方法的时候,就返回辅助栈栈顶的值即可。

源码:

import java.util.Stack;public class Test1 {    public class Solution {        Integer temp = null;        Stack<Integer> min = new Stack<Integer>();//此为辅助栈        Stack<Integer> data = new Stack<Integer>();//此为主栈        public void push(int node) {            if (temp!=null) {//如果栈中已经有数据                if(node <= temp){//和之前的最小值进行比较                    min.push(node);//若node不大于之前的最小值,就将之push到min栈中                    temp = node;                }                data.push(node);//进行主栈基础的数据push            }else {//若栈中没有数据                temp = node;                data.push(node);                min.push(node);            }        }               public void pop() {            int num = min.pop();            if(data.pop()!=num){min.push(num);}//此处主栈已经进行了pop操作        }        public int top() {            int num = data.peek();            return num;        }        public int min() {            int num = min.peek();//获取min栈栈顶元素即可            return num;        }    }}

运行测试:

第一种:
运行时间:15ms
占用内存:8628k

总结:
要抓住题目的核心,使用任何数据结构进行辅助的操作都可以。另外也可使用ArrayList来做辅助存储的容器。但是借助栈来实现总是要方便一些的。可以少处理很多不必要的逻辑。

原创粉丝点击