包含min函数的栈

一、题目描述

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

二、解题思路

用一个辅助栈存储当前栈中的最小元素值。

三、Java代码实现

import java.util.Stack;

public class Solution {
    Stack<Integer> stack = new Stack<Integer>();
    Stack<Integer> minStack = new Stack<Integer>();
    int min = Integer.MAX_VALUE;

    public void push(int node) {
        stack.push(node);
        if(node < min)
            min = node;
        minStack.push(min);
    }

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

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

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