剑指offer--包含min函数的栈

描述

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

代码

定义两个栈s1，s2。s1是自然栈，也就是正常的压栈和出栈，而s2是保存当前最小值的栈，当有一个元素压栈时，首先比较s2的栈顶元素和该元素的大小，如果该元素小于s2的栈顶元素，则向s2压栈（更新一下最小值）。当弹栈的时候，也要比较s1和s2的栈顶元素，如果相等，需要将s1和s2的栈顶元素都弹出。

import java.util.Stack;

public class Solution {

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

    public void pop() {
        int v=s1.pop();
        if(v==s2.peek()){
            s2.pop();
        }
    }

    public int top() {
        return s1.peek();
    }

    public int min() {
        return s2.peek();
    }
}