【剑指offer】20包含min函数的栈

程序猿周周

于 2018-03-18 15:29:38 发布

阅读量249

点赞数

CC 4.0 BY-SA版权

分类专栏：剑指Offer 剑指offer 文章标签：数据结构栈

本文链接：https://blog.youkuaiyun.com/adminpd/article/details/79601025

剑指offer 同时被 2 个专栏收录

38 篇文章

订阅专栏

剑指Offer

37 篇文章

订阅专栏

本文介绍了一种特殊栈的数据结构实现，该栈包含一个min函数用于返回栈中最小元素。通过使用辅助栈来跟踪最小值，确保了min函数的时间复杂度为O(1)。文章提供了完整的Java代码示例。

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

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

import java.util.Stack;

public class Solution {
/*思路：
利用一个辅助栈来存放最小值
    栈  3，4，2，5，1
    辅助栈 3，3，2，2，1
每入栈一次，就与辅助栈顶比较大小，如果小就入栈，如果大就入栈当前的辅助栈顶
当出栈时，辅助栈也要出栈
这种做法可以保证辅助栈顶一定都当前栈的最小值
*/
    Stack<Integer> data = new Stack<Integer>();
    Stack<Integer> min = new Stack<Integer>();

    public void push(int node) {
        data.push(node);
        if(min.isEmpty())
            min.push(node);
        else{
            if(node<=min.peek())
                min.push(node);
        }
    }

    public void pop() {
        if(data.peek()==min.peek())
            min.pop();
        data.pop();
    }

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

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

    public static void main(String[] args) {
        Solution s=new Solution();

        s.push(2);
        s.push(1);
        s.push(3);
        s.push(4);

        System.out.println(s.min());
        System.out.println(s.top());
    }
}