前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >[算法题] 实现一个特殊的栈,在基本栈的基础上,增加返回栈中最小元素的方法

[算法题] 实现一个特殊的栈,在基本栈的基础上,增加返回栈中最小元素的方法

作者头像
CoderJed
发布2019-05-14 14:25:18
7380
发布2019-05-14 14:25:18
举报
文章被收录于专栏:Jed的技术阶梯Jed的技术阶梯

1. 题目要求

实现一个特殊的栈,在栈的基本功能的基础上,增加一个功能:返回栈中最小元素 要求:

  • pop(),push(),getMin()操作的复杂度都为O(1)
  • 设计的栈类型可以使用现成的栈结构

2. 思路1

3. 思路1 Java代码实现

代码语言:javascript
复制
public static class MyStack1 {

        private Stack<Integer> dataStack;
        private Stack<Integer> minStack;

        public MyStack1() {
            this.dataStack = new Stack<>();
            this.minStack = new Stack<>();
        }

        public int getMin() {
            if(minStack.isEmpty()) {
                throw new RuntimeException("Your stack is empty!");
            }
            return minStack.peek();
        }

        public void push(int element) {
            if(minStack.isEmpty()) {
                minStack.push(element);
            } else if(element <= getMin()) {
                minStack.push(element);
            } else {
                int min = minStack.peek();
                minStack.push(min);
            }
            dataStack.push(element);
        }

        public int pop() {
            if(dataStack.isEmpty()) {
                throw new RuntimeException("Your stack is empty!");
            }
            minStack.pop();
            return dataStack.pop();
        }

}

4. 思路2

思路2对思路1进行了空间上的优化,在思路1中可能会压入重复的元素,优化思路如下:

5. 思路2 Java代码实现

代码语言:javascript
复制
public static class MyStack2 {

        private Stack<Integer> dataStack;
        private Stack<Integer> minStack;

        public MyStack2() {
            this.dataStack = new Stack<>();
            this.minStack = new Stack<>();
        }

        public int getMin() {
            if(minStack.isEmpty()) {
                throw new RuntimeException("Your stack is empty!");
            }
            return minStack.peek();
        }

        public void push(int element) {
            if(minStack.isEmpty()) {
                minStack.push(element);
            } else if(element <= getMin()) {
                minStack.push(element);
            } // 只有当push的元素小于minStack的栈顶元素时才minStack才push
            dataStack.push(element);
        }

        public int pop() {
            if(dataStack.isEmpty()) {
                throw new RuntimeException("Your stack is empty!");
            }
            int value = dataStack.pop();
            // 只有dataStack的栈顶元素=minStack的栈顶元素时,minStack才弹出
            if(value == getMin()) {
                minStack.pop();
            }
            return value;
        }

}
本文参与 腾讯云自媒体分享计划,分享自作者个人站点/博客。
原始发表:2019.05.02 ,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 作者个人站点/博客 前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体分享计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
目录
  • 1. 题目要求
  • 2. 思路1
  • 3. 思路1 Java代码实现
  • 4. 思路2
  • 5. 思路2 Java代码实现
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档