剑指Offer之包含min函数的栈-优快云博客

本文链接：https://blog.youkuaiyun.com/weixin_40100141/article/details/79498671

本文介绍了一种特殊栈的设计方案，通过维护两个栈来确保能够高效地获取栈中的最小元素。主要思路是在进行压栈操作时，同时更新辅助栈以保持其栈顶为当前最小值，弹栈时依据条件同步更新这两个栈。

题目描述
定义栈的数据结构，请在该类型中实现一个能够得到栈最小元素的min函数。

思路：
1、借助一个辅助栈stack2，这个栈内存放着栈最小元素的值，并且栈顶元素的值永远是最小的。

2、入栈时，需要判断stack2的栈顶元素是否大于要入栈的值，若是大于，则将值入至stack2。

3、出栈时，需要判断栈顶的值与辅助栈顶的值是否相等，相等则一起出栈，否则只有栈顶的值出栈。

4、求min()时只需要将辅助栈顶peek()就OK了。

贴代码：

import java.util.Stack;

public class Solution {
//借助一个辅助栈
    Stack<Integer> stack1 = new Stack<>();
    Stack<Integer> stack2 = new Stack<>();//栈顶的数永远是最小的数

    //当压栈时，需要判断压栈的数是否比辅助栈中栈顶的数小
    public void push(int node) {
        stack1.push(node);
        if(stack2.isEmpty()){
            stack2.push(node);
        }
        else{
            //压栈数据比栈内最小值小，则往stack2压栈
            if(node <= stack2.peek()){
                stack2.push(node);
            }
        }
    }
    //弹栈的时候如果栈顶元素和最小值栈元素相等则一起弹栈，否则只是栈顶弹栈
    public void pop() {
        if(stack1.isEmpty()){
            return;
        }
        if(stack1.peek() == stack2.peek()){
            stack2.pop();
        }
        stack1.pop();
    }

    public int top() {
        if(!stack1.isEmpty()){
            return stack1.peek();
        }
        return 0;
    }

    public int min() {
        if(!stack2.isEmpty()){
            return stack2.peek();
        }
        return 0;
    }
}