【牛客 - 剑指offer】JZ30 包含min函数的栈 Java实现


剑指offer题解汇总 Java实现

https://blog.csdn.net/guliguliguliguli/article/details/126089434

本题链接

https://www.nowcoder.com/exam/oj/ta?tpId=13

题目

在这里插入图片描述
在这里插入图片描述

思路 & 代码

在这里插入图片描述

栈结构的push、pop、top操作都是O(1)、O(1)、O(1),但是min函数做不到,于是想到在push的时候就将最小值记录下来,由于栈先进后出的特殊性,我们可以构造一个单调栈,保证栈内元素都是递增的,栈顶元素就是当前最小的元素。

创建两个栈:

  • stack1用来记录实际添加的数据

  • stack2一个用来记录当前栈中的最小值,是一个单调栈

    • 当stack1为空,说明目前栈中没有数据,则添加进来的数据就是当前所有数据的最小值,stack2中添加该数据

    • 当stack1不为空,说明栈中有数据,用stack2栈顶元素与即将添加的数据node进行比较

      • 如果栈顶元素值大于node,则说明当前node是栈中最小值,stack2中添加node数据

      • 如果栈顶元素值小于node,则说明stack2栈顶元素的值是栈中最小值,则再次添加stack2的栈顶元素到stack2中

import java.util.*;
import java.util.Stack;

public class Solution {

    //记录正常的进栈、出栈数据
    private Stack<Integer> stack1 = new Stack<>();

    //构造一个单调栈
    private Stack<Integer> stack2 = new Stack<>();


    public void push(int node) {
        stack1.push(node);
        //如果stack2为空,则直接把node加进去
        if (stack2.isEmpty()) {
            stack2.push(node);
        } else {
            //stack2不为空,将node与stack2栈顶元素相比较
            if (node > stack2.peek()) {
                //node较大,则重复添加stack2的栈顶元素
                stack2.push(stack2.peek());
            } else {
                //node较小,添加node
                stack2.push(node);
            }
        }
    }

    public void pop() {
        stack1.pop();
        stack2.pop();
    }

    public int top() {
        return stack1.peek();
    }

    public int min() {
        return stack2.peek();
    }
}

版权声明:本文为guliguliguliguli原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。