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();
    }
}