1栈和队列--设计一个有getMin功能的栈

最新推荐文章于 2020-06-08 22:04:26 发布

原创最新推荐文章于 2020-06-08 22:04:26 发布 · 358 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#栈

算法与数据结构题目最优解专栏收录该内容

8 篇文章

订阅专栏

本文介绍了一种使用两个栈实现带有获取最小元素功能的数据结构的方法。通过维护一个辅助栈来跟踪最小值，该方法能够在常数时间内完成压栈、弹栈及获取最小值的操作。

实现功能：
- 正常压栈
- 正常弹栈
- 返回栈中最小值

方法1：
这里写图片描述

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();
        }

方法2：
这里写图片描述

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

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

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