【2】包含min函数的栈

题目描述

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

解题思路

1、首先定义两个数据栈 stack 和 min ，其中stack用于存放所有的数据，min用于存放每次push的最小值。
2、push函数：直接push进stack栈，而min栈，若是空栈直接push，若不是空栈，判断push的数据与min栈的栈顶元素谁小，放入小的那个元素。
3、pop函数：每次pop的时候，两个数据栈都要pop出元素，以确保min栈顶的最小值存在于stack栈中。
4、min函数：用来得到min栈顶元素，即最小值
5、top函数：目前不知道这个函数用来干什么的，先留个坑，之后了解了回来填。

图解

import java.util.Stack;

public class Solution {

    //存放数据的栈
    Stack<Integer> stack = new Stack<>();
    //存放最小值的栈
    Stack<Integer> min = new Stack<>();
  
    public void push(int node) {
      stack.push(node);
      if(min.isEmpty()){
        min.push(node);
      }else{
        min.push(node>min.peek()?min.peek():node);
      }
    }
    
    public void pop() {
      //每次弹出，两个栈都弹出数据，以确保min栈最小元素存在于stack栈中
      if(!stack.isEmpty()){
        stack.pop();
        min.pop();
      }else{
        System.out.println("栈已空");
      }
    }
    
    public int top() {
        return stack.peek();
    }
    
    //得到min栈的最小值
    public int min() {
        return min.peek();
    }
}