剑指20.包含min函数的栈

题目描述

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

 

思路

要求实现一个O(1)时间复杂度的返回最小值的栈。正常情况下，栈的push，pop操作都为O(1),但是返回最小值，需要遍历整个栈，时间复杂度为O(n)，所以这里需要 空间换时间 的思想。
方法：使用辅助栈存放最小值。

    栈  3，4，2，5，1

    辅助栈 3，3，2，2，1

每入栈一次，就与辅助栈顶比较大小，如果小就入栈，如果大就入栈当前的辅助栈顶

当出栈时，辅助栈也要出栈

这种做法可以保证辅助栈顶一定都当前栈的最小值。

 

参考代码

import java.util.Stack;

public class Solution {
    Stack<Integer> stackData = new Stack<Integer>(); // 数据栈
    Stack<Integer> stackMin = new Stack<Integer>(); // 辅助栈
    public void push(int node) {
        stackData.push(node);
        if(stackMin.isEmpty() || node < stackMin.peek())
            stackMin.push(node);
        else
            stackMin.push(stackMin.peek());
    }
    
    public void pop() {
        stackData.pop();
        stackMin.pop();
    }
    
    public int top() {
        return stackData.peek();
    }
    
    public int min() {
        return stackMin.peek();
    }
}