数据结构篇-02:最小栈

 对于这道题,除了 getMin 外的功能,传统的 栈 结构中都有,所以重点在于如何实现 getMin 方法。

有两类方法:使用辅助栈/不使用辅助栈

使用辅助栈的解法一

定义一个 栈 来实现常规功能,另外定义一个栈(最小栈)专门用于实现 getMin 方法

使得 [最小栈] 的栈顶始终都是当前栈中的最小元素即可

class MinStack {
    private Stack<Integer> dataStack;
    private Stack<Integer> minStack;
    
    public MinStack() {
        dataStack = new Stack<>();
        minStack = new Stack<>();
    }
    
    public void push(int x) {
        // 先将 x 压入数据栈
        dataStack.push(x);
        // 如果 x 是当前的最小值,则也需要将 x 压入辅助栈
        if (minStack.isEmpty() || x <= minStack.peek()) {
            minStack.push(x);
        }
    }
    
    public void pop() {
        // 先删除数据栈的栈顶元素 x
        int x = dataStack.pop();
        // 若 x 是当前的最小值,则也需要删除辅助栈的栈顶元素
        if (x == minStack.peek()) {
            minStack.pop();
        }
    }
    
    public int top() {
        return dataStack.peek();
    }
    
    public int getMin() {
        return minStack.peek();
    }
}

 不使用辅助栈的解法一

不使用辅助栈的话,就需要额外保存当前的最小值

定义一个节点类Node,里面除了保存有当前元素外,还包含当前的最小元素。这样当需要最小元素时,直接弹出栈顶元素,然后获取栈顶元素中的最小值即可。 

class MinStack {

    private Stack<Node> stack;

    public MinStack() {
        stack = new Stack<>();
    }
    
    public void push(int x) {
        // 将元素值 x 和 当前最小值 同时入栈。
        if (stack.isEmpty()) {
            stack.push(new Node(x, x));
        } else {
            stack.push(new Node(x, Math.min(x, stack.peek().min)));
        }
    }
    
    public void pop() {
        stack.pop();
    }
    
    public int top() {
        // 返回栈顶元素中的元素值。
        return stack.peek().val;
    }
    
    public int getMin() {
        // 返回栈顶元素的最小值
        return stack.peek().min;
    }

    private static class Node {
        int val;
        int min;

        public Node(int val, int min) {
            this.val = val;
            this.min = min;
        }
    }
}

相关推荐

  1. 数据结构(代码

    2024-01-28 02:28:02       55 阅读
  2. 03 数据结构

    2024-01-28 02:28:02       40 阅读

最近更新

  1. docker php8.1+nginx base 镜像 dockerfile 配置

    2024-01-28 02:28:02       94 阅读
  2. Could not load dynamic library ‘cudart64_100.dll‘

    2024-01-28 02:28:02       101 阅读
  3. 在Django里面运行非项目文件

    2024-01-28 02:28:02       82 阅读
  4. Python语言-面向对象

    2024-01-28 02:28:02       91 阅读

热门阅读

  1. python的深浅拷贝

    2024-01-28 02:28:02       58 阅读
  2. HTTP协议(简单知识点)

    2024-01-28 02:28:02       57 阅读
  3. 【js大坑】苹果ios的日期时间兼容问题

    2024-01-28 02:28:02       51 阅读
  4. 《Leetcode》动态规划-求不相邻数的最大和

    2024-01-28 02:28:02       57 阅读
  5. 加固安全防线:解决常见漏洞的实用指南

    2024-01-28 02:28:02       55 阅读