1 题目描述

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

2 解题思路

构造三个栈，一个栈存放正常数据节点，一个栈来存放最小元素栈，一个栈来存放最大元素栈。

• 入栈：将每次入栈的数字和最小元素栈的栈顶元素相比 如果入栈元素小于等于最小元素栈的栈顶元素，那么将该元素入数据栈和最小元素栈

• 出栈：将每次出栈的元素和最小元素栈的栈顶元素相比 ，如果相等那么最小元素栈和数据节点栈的栈顶元素都出栈，否则只有数据栈 栈顶元素出栈。

3 代码实现

import java.util.Stack;

public class Solution {


   Stack<Integer> data=new Stack<Integer>();
   Stack<Integer> min=new Stack<Integer>();
   Stack<Integer> max=new Stack<Integer>();
   public void push(int node) {
       
       if(min.size()==0){
           min.push(node);
       }
       if(max.size()==0){
           max.push(node);
       }
       if(min.peek()>=node){
           min.push(node);
       }
       if(max.peek()<=node){
           max.push(node);
       }
       
       data.push(node);
   }
   
   public void pop() {
       
       int node=data.pop();
       
       if(min.peek()==node){
           min.pop();
       }
       if(max.size()>0 && max.peek()==node){
           max.pop();
       }
       
   }
   
   public int top() {
       
       return max.peek();
   }
   
   public int min() {
       
       return min.peek();
       
   }
}