【Leetcode】155.最小栈

最新推荐文章于 2024-08-10 22:52:22 发布

沉着的码农

最新推荐文章于 2024-08-10 22:52:22 发布

阅读量48

点赞数 1

分类专栏： LeetCode 文章标签： leetcode 算法 java

本文链接：https://blog.csdn.net/m0_72670269/article/details/130230598

版权

LeetCode 专栏收录该内容

16 篇文章 0 订阅

订阅专栏

文章介绍了如何使用两个栈实现一个最小栈的功能，每次入栈时会将当前数值与min栈的栈顶元素比较，如果小于或等于栈顶元素，则将数值压入min栈。在弹出元素时，如果弹出的元素是当前栈的最小值，同时也要从min栈中弹出。这样，min栈的栈顶始终保持着栈中最小值。

摘要由CSDN通过智能技术生成

题目

在这里插入图片描述

思路

我们解决这道题只需要再创建一个栈来存放每次入栈的最小值

** 例如 **
在这里插入图片描述
每次入min栈时和min栈顶元素进行比较
这个时候栈顶存放的就是stack栈中数据的最小值

当我们弹出stack栈中的数据时如果这个数据和min栈栈顶元素相同此时把min栈栈顶的元素也弹出此时min中的栈顶元素还是stack中的最小元素

代码

class MinStack {
	//初始化两个栈
    Stack<Integer> stack;
    Stack<Integer> min;
    public MinStack() {
        stack = new Stack<>();
        min = new Stack<>();
    }
    
    //压栈方法
    public void push(int val) {
    	//先将val压入stack中
        stack.push(val);
        //如果此时min栈为空 说明这是第一个元素 直接插入不需要判断
        if(min.empty()){
            min.push(val);
        }else{
        	//如果此时min栈不为空 则和栈顶元素进行判断
            if(val <= min.peek()){
                //必须有等于
                //因为我们在pop时 如果是最小值要将两个栈的栈顶元素都弹出
                //如果stack中恰好插入了两个相同的最小元素
                //此时如果pop 则min栈的栈顶就不是最小的值了
                min.push(val);
            }
        }
    }
    
    public void pop() {
    	//如果栈为空 则直接返回
        if(stack.empty()){
            return;
        }
        //取出stack栈顶元素 
        int k = stack.pop();
        //判断是否和min栈顶元素相同 
        //如果相同min也弹出
        if(k == min.peek()){
            min.pop();
        }
    }
    //获取stack栈顶元素
    public int top() {
    	//如果栈为空返回0
        if(stack.empty()){
            return 0;
        }
        return stack.peek();
    }
    //获取栈中的最小值
    //我们只需要返回min栈栈顶的元素即可
    public int getMin() {
        if(min.empty()){
            return 0;
        }
        return min.peek();
    }
}