[算法练习-剑指offer]题20.包含min函数的栈（Java）

题目

题号:20
题目名:包含min函数的栈

编程语言

Java

题目描述

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

初次思路

题意是要获取栈的最小值,时间复杂度应为O（1）,说明可以直接获取值.栈的数据操作类型是FILO,要再设置一个数列来排序有些麻烦,这里不妨使用空间换时间,将要存入栈的值和最小值一起存

push的时候直接将对应node和最小值min一起存进去

pop就将node和min一块出栈

top就是获取栈顶的min下面一个值

getmin就是一起存入的min

解题代码

public class Solution {
    /**
     * 这里可以使用空间换时间
     * 每次压栈两个,一个是当前值,一个是当前最小值
     * 取最小值只要取最新压进去的值就可以了
     */
    Stack<Integer> stack = new Stack<>();

    public void push(int node) {
        //如果当前栈为空,直接将当前值和最小值压入
        if (stack.isEmpty()) {
            stack.push(node);
            stack.push(node);
        }else {
            //否则,比较最小值
            Integer min = stack.peek();
            if(node<min) {
                //如果比最小值小,将当前值压入栈
                stack.push(node);
                stack.push(node);
            }else {
                stack.push(node);
                stack.push(min);
            }
        }
    }

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

    public int top() {
        return stack.get(stack.size()-2);
    }

    public int min() {
        //考虑使用空间换时间
        return stack.peek();
    }
}