BM43 包含min函数的栈

1.题目描述

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

此栈包含的方法有：

push(value):将value压入栈中

pop():弹出栈顶元素

top():获取栈顶元素

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

数据范围：操作数量满足 0≤n≤300 0≤n≤300  ，输入的元素满足 ∣val∣≤10000 ∣val∣≤10000 
进阶：栈的各个操作的时间复杂度是 O(1) O(1)  ，空间复杂度是 O(n) 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

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

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

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

示例1

输入：

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

返回值：

-1,2,1,-1

2.解题思路

为了记录当前栈中的最小值，我们需要引入另一个栈stack2,当push操作要加入栈中的node小于stack2的栈顶元素或者stack2为空时，说明当前这个node是在stack1中最小的值，要把它加入到stack2中，另外在stack1要弹出栈顶元素时，仍需要判断当前弹出的元素是否与stack2栈顶元素相同，如果相同，也要把记录最小值栈stack2栈顶元素一同弹出，这是一种以空间换时间的方式，保证所有操作都能在O(1)的时间复杂度上完成。

3.代码实现

import java.util.*;
import java.util.Stack;

public class Solution {

    public Stack<Integer> stack1 = new Stack<>();

    public Stack<Integer> stack2 = new Stack<>();
    
    public void push(int node) {
        stack1.push(node);
        if (stack2.isEmpty() || node <= stack2.peek()) {
            stack2.push(node);
        }
        return;
        
    }
    
    public void pop() {
        int val = stack1.pop();
        if (val == stack2.peek()) {
            stack2.pop();
        }
        return;
    }
    
    public int top() {
        return stack1.peek();
    }
    
    public int min() {
        return stack2.peek();
    }
}