美文网首页
实现一个特殊的栈, 在实现栈的基本功能的基础上, 再实现返 回栈

实现一个特殊的栈, 在实现栈的基本功能的基础上, 再实现返 回栈

作者: shoulda | 来源:发表于2018-07-15 21:49 被阅读0次

题目
实现一个特殊的栈, 在实现栈的基本功能的基础上, 再实现返
回栈中最小元素的操作。
【要求】:
1.pop、 push、 getMin操作的时间复杂度都是O(1)
2.设计的栈类型可以使用现成的栈结构

解答:用两个栈,一个数据栈,一个记录最小值栈

public static class MyStack1 {
        private Stack<Integer> stackData;
        private Stack<Integer> stackMin;

        public MyStack1() {
            this.stackData = new Stack<Integer>();
            this.stackMin = new Stack<Integer>();
        }

        public void push(int newNum) {
            if (this.stackMin.isEmpty()) {
                this.stackMin.push(newNum);
            } else if (newNum <= this.getmin()) {
                this.stackMin.push(newNum);
            }
            this.stackData.push(newNum);
        }

        public int pop() {
            if (this.stackData.isEmpty()) {
                throw new RuntimeException("Your stack is empty.");
            }
            int value = this.stackData.pop();
            if (value == this.getmin()) {
                this.stackMin.pop();
            }
            return value;
        }

        public int getmin() {
            if (this.stackMin.isEmpty()) {
                throw new RuntimeException("Your stack is empty.");
            }
            return this.stackMin.peek();
        }
    }

相关文章

网友评论

      本文标题:实现一个特殊的栈, 在实现栈的基本功能的基础上, 再实现返 回栈

      本文链接:https://www.haomeiwen.com/subject/phlmpftx.html