【栈队列堆】【打卡80天】《剑指Offer》2刷：JZ30 包含min函数的栈-CSDN博客

本文链接：https://blog.csdn.net/Sunshineoe/article/details/121329890

1、题目描述

定义栈的数据结构，请在该类型中实现一个能够得到栈中所含最小元素的 min 函数，输入操作时保证 pop、top 和 min 函数操作时，栈中一定有元素。

此栈包含的方法有：

push(value):将value压入栈中

pop():弹出栈顶元素

top():获取栈顶元素

min():获取栈中最小元素

数据范围：操作数量满足，输入的元素满足 $|val| \le 10000 \$
进阶：栈的各个操作的时间复杂度是 $O(1)\$ ，空间复杂度是 $O(n)\$

示例:

输入: ["PSH-1","PSH2","MIN","TOP","POP","PSH1","TOP","MIN"]

输出: -1,2,1,-1

解析:

"PSH-1"表示将-1压入栈中，栈中元素为-1

"PSH2"表示将2压入栈中，栈中元素为2，-1

“MIN”表示获取此时栈中最小元素==>返回-1

"TOP"表示获取栈顶元素==>返回2

"POP"表示弹出栈顶元素，弹出2，栈中元素为-1

"PSH-1"表示将1压入栈中，栈中元素为1，-1

"TOP"表示获取栈顶元素==>返回1

“MIN”表示获取此时栈中最小元素==>返回-1

2、算法分析

注意一：

push(value):将value压入栈中

pop():弹出栈顶元素 stack.pop()，弹出栈顶元素

top():获取栈顶元素 stack.peek()，获取栈顶元素值，但是不弹出元素

min():获取栈中最小元素 stack.peek()，获取栈顶元素，不弹出

注意二：

输入操作时保证 pop、top 和 min 函数操作时，栈中一定有元素。栈始终不是Empty。

本题使用两个栈，一个栈正常存储元素，另一个栈的栈顶存储始终是整个栈中最小的元素。

3、代码实现

import java.util.Stack;

public class Solution{
    Stack<Integer> stackData = new Stack<Integer>();
    Stack<Integer> stackMin = new Stack<Integer>();

    public void push(int node){
        stackData.push(node);
        if(stackMin.isEmpty()){
            stackMin.push(node);
        }else{
            stackMin.push(stackMin.peek());
        }
        
        if(stackMin.peek() > node){
            stackMin.push(node);
        }
    }

    // 弹出栈顶元素
    public void pop(){
        if(!stackData.isEmpty()){
            stackData.pop();
        }
        if(!stackMin.isEmpty){
            stackMin.pop();
        }
    }

    public int top(){
        if(!stackData.isEmpty()){
            stackData.peek();
        }
    }

    public int min(){
        if(!stackMin.isEmpty()){
            stackMin.peek();
        }
    }
}