《剑指offer》刷题笔记：20-包含min函数的栈

题目描述

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

解题思路

新建两个栈 s1，s2，利用辅助栈 s2 来存放最小值
s1 每入栈一个元素，入栈元素就与 s2 栈顶比较大小：
1、如果入栈元素小于或等于 s2 栈顶，则将该元素压入 s2；
2、否则将当前最小元素（即 s2 栈顶）再次压入 s2。
这种做法可以保证辅助栈顶一定都当前栈的最小值，并且两个栈大小相同（同步栈）

代码实现

import java.util.Stack;

public class Solution {

    Stack<Integer> s1 = new Stack<>();
    Stack<Integer> s2 = new Stack<>();
    public void push(int node) {
        s1.push(node);
        if(s2.empty()){
            s2.push(node);
        }
        else{
            if(node <= s2.peek()){
                s2.push(node);
            }
            else
                s2.push(s2.peek());
        }
    }
    
    public void pop() {
        s1.pop();
        s2.pop();
    }
    
    public int top() {
        return s1.peek();
    }
    
    public int min() {
        return s2.peek();
    }
}