获取栈中最小值

梦断泪影

于 2022-03-16 16:11:46 发布

阅读量416

点赞数

CC 4.0 BY-SA版权

文章标签： java 开发语言

本文链接：https://blog.youkuaiyun.com/mengduan913/article/details/123529088

本文介绍了一个Java实现的MyStack类，用于维护一个栈的同时，能够动态获取栈中的最小值。通过构造器初始化两个栈，一个用于存储实际数据，另一个记录当前栈中的最小值。关键方法包括push、pop和getMin，实现了高效查找和操作最小值。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

import java.util.Stack;

/**
 * @author chencc
 * @Description 获取栈中最小值
 * @Date 2022/3/16 15:06
 */
public class GetMinStack {
    //构建一个栈
    public static class MyStack {
        //真实栈用来存放数据
        public Stack<Integer> stackData;
        //存在当前栈数据中,最小数
        public Stack<Integer> stackMin;

        //构造器,初始化
        public MyStack() {
            this.stackData = new Stack<Integer>();
            this.stackMin = new Stack<Integer>();
        }

        //插入数据
        public void push(int value) {
            //当最小栈为空时,直接插入当前数
            if (stackMin.isEmpty()) {
                stackMin.push(value);

                //当插入数小于当前栈顶的数,直接插入当前数
            } else if (value < stackMin.peek()) {
                stackMin.push(value);

                //此时插入当前栈顶的数
            } else {
                stackMin.push(stackMin.peek());
            }
            stackData.push(value);
        }

        //弹出栈数据
        public int pop() {
            if (stackData.isEmpty()) {
                throw new RuntimeException("当前栈为空");
            }
            stackMin.pop();
            return stackData.pop();
        }

        //获取栈中最小值
        public int getMin() {
            if (stackMin.isEmpty()) {
                throw new RuntimeException("当前栈为空");
            }
            return stackMin.peek();
        }
    }

    public static void main(String[] args) {
        MyStack stack = new MyStack();
        stack.push(3);
        stack.push(0);
        System.out.println(stack.getMin());
        stack.pop();
        System.out.println(stack.getMin());
    }
}