包含min函数的栈

最新推荐文章于 2023-07-24 10:28:56 发布

风兮水寒

最新推荐文章于 2023-07-24 10:28:56 发布

阅读量321

点赞数

CC 4.0 BY-SA版权

本文链接：https://blog.youkuaiyun.com/wjb857999880/article/details/51236306

本文介绍了一种特殊栈的数据结构实现方法，在该结构中加入了一个能获取当前栈内最小元素的min函数。通过维护两个栈，一个是数据栈用于存储实际数据，另一个是辅助栈专门用来保存最小值，确保了min函数能在常数时间内返回结果。

题目描述

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

思路：

定义一个保存最小值的辅助栈。

代码：

import java.util.Stack;

public class Solution {

    private Stack<Integer> minStack = new Stack<Integer>();
    private Stack<Integer> dataStack = new Stack<Integer>();

    public void push(int node) {
        dataStack.push(node);
        if(minStack.empty()||minStack.peek()>node)
            minStack.push(node);
        else
            minStack.push(minStack.peek());
    }

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

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

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