包含min函数的栈-剑指Offer（Java语言）-CSDN博客

本文链接：https://blog.csdn.net/ChaunceyChen/article/details/88090287

本文介绍了一种特殊的栈数据结构，通过辅助栈实现O(1)时间复杂度的最小元素查询。主要讨论了如何在入栈和出栈操作中维护辅助栈，确保其栈顶始终为原栈中的最小值。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目描述

定义栈的数据结构，请在该类型中实现一个能够得到栈中所含最小元素的min函数（时间复杂度应为O（1））。

思路

使用一个辅助栈来存储最小的元素，辅助栈栈顶为原栈中最小的元素。

入栈：将元素压入原栈，如果该元素比辅助栈栈顶元素小，则压入辅助栈。

出栈：原栈栈顶元素出栈，如果该出栈的元素与辅助栈栈顶元素相同，则辅助栈也出栈。

min函数：获取辅助栈栈顶

为什么只需要在辅助栈中压入小于辅助栈栈顶的元素？

可能会想到一个问题：假设当前辅助栈栈顶为8，现在要在原栈压入9，虽然现在原栈中的最小值是8，但是如果原栈中的其他元素都大于9，那么当原栈中8被出栈时，最小值不是变成了9么？所以是不是应该把9也保留下来，并放在辅助栈8的下面？

解答：不用。因为在原栈中，8一定是在9之前入栈的，所以当8出栈的时候，9早已经出栈了，即意味着最小值不会是这个9.

代码：

import java.util.Stack;

public class Solution {

    Stack<Integer> m=new Stack<>();
    Stack<Integer> s=new Stack<>();//辅助栈，栈顶为最小元素
    public Solution(){
        m.push(Integer.MAX_VALUE);
    }
    public void push(int node) {
        s.push(node);
        if(node<m.peek()){
            m.push(node);
        }
    }
    
    public void pop() {
        int t=s.pop();
        if(t==m.peek()){
            m.pop();
        }
    }
    
    public int top() {
        return s.peek();
    }
    
    public int min() {
        return m.peek();
    }
}