每天一道剑指offer-包含min函数的栈-优快云博客

本文介绍了一种特殊栈的数据结构实现，通过辅助栈来跟踪并返回栈内最小元素，确保min函数的时间复杂度为O(1)。文章详细解释了如何在元素入栈和出栈时维护这个辅助栈，以保持最小值的实时更新。

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

题目描述

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

思路

定义一个辅助的成员变量，当有新元素进栈时，比较成员变量和新进栈元素的大小，如果新进栈元素小，则更新成员变量中值，但当栈中元素出栈时，无法得到次最小的元素，所以仅定义一个成员变量的值存放最小的值时不够的，还需要借助一个辅助栈，把每次新入栈元素的最小值都给保存下来。
在这里插入图片描述

代码

public class Solution {

    Stack<Integer> stack1 = new Stack<>();
    Stack<Integer> stack2 = new Stack<>();
    public void push(int node) {
        stack1.push(node);
        if(stack2.empty()){
            stack2.push(node);
        }else{ 
            if(node<=(int)stack2.peek()){  //比较新入栈元素（node)和辅助栈中最小元素（(int)stack2.peek()）
            stack2.push(node);
        }else{
                stack2.push(stack2.peek());
            } 
                }
    }
    
    public void pop() {
       stack2.pop();
       stack1.pop();
    }
    
    public int top() {
        return (int)stack1.peek();
    }
    
    public int min() {
        return (int)stack2.peek();
    }
}