面试题21：包含min函数的栈

题目描述

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

代码实现

import java.util.Stack;

public class Solution {

    Stack stack = new Stack();
    Stack min = new Stack();
    
    public void push(int node) {
        stack.push(node);
        if(min.isEmpty() || node <= min.peek())
            min.push(node);
        else
            min.push(min.peek());
    }
    
    public void pop() {
        stack.pop();
        min.pop();
    }
    
    public int top() {
        return stack.peek();
    }
    
    public int min() {
        return min.peek();
    }
}

主要思路

1、这道题需要用到两个栈，一个是数据栈，一个是辅助栈
2、每次入栈的时候，辅助栈如果是空的，或者本次入栈的数据不大于栈顶，直接入栈，比如2,1,1（数据栈）和2,1,1（辅助栈）；否则压入的是当前栈顶的复制值，比如2,3,4（数据栈）和2,2,2（辅助栈），从而保证数据栈里面的最小值永远在辅助栈的栈顶
3、需要注意的是，实例化栈的时候，一定要传入一个泛型参数（本题中的栈存储的只能是int类型的值），否则比较大小的时候会出错（node <= min.peek()）
4、访问栈顶元素用peek方法，而不是top方法