20包含min函数的栈

 

题目描述

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

 

建一个辅助栈，把每次最小最小的元素（之前最小元素，与当前新入栈的元素比较）放在辅助栈里。

 

 import java.util.Stack;

 public class Solution {
     Stack<Integer> stack = new Stack<Integer>();
     Stack<Integer> min = new Stack<Integer>();
     public void push(int node) {
         if(stack.empty()){
             stack.push(node);
             min.push(node);
         }
         else{
             if(node < min.peek())
                 min.push(node);
             else
                 min.push(min.peek());
             stack.push(node);
             }

     }

     public void pop() {
         stack.pop();
         min.pop();
     }

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

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