剑指offer之包含min函数的栈（Java实现）

最新推荐文章于 2023-04-20 17:32:16 发布

原创最新推荐文章于 2023-04-20 17:32:16 发布 · 423 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#剑指offer

剑指offer 专栏收录该内容

65 篇文章

订阅专栏

本文介绍了一种特殊栈的数据结构实现，通过辅助栈来跟踪并返回栈内最小元素，确保min函数的时间复杂度为O(1)。利用Java语言详细展示了如何在栈中加入、删除元素，并始终保持对当前最小值的快速访问。

包含min函数的栈

NowCoder

题目描述:

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

解题思路：

import java.util.Stack;
import java.util.Arrays;
public class Solution {
//借用辅助栈存储min的大小，自定义了栈结构
    private int size;
    private int min = Integer.MAX_VALUE;
    private Stack<Integer> minStack = new Stack<Integer>();
    private Integer[] elements = new Integer[10];
    public void push(int node) {
        ensureCapacity(size+1);
        elements[size++] = node;
        if(node <= min){
            minStack.push(node);
            min = minStack.peek();
        }else{
            minStack.push(min);
        }
    //    System.out.println(min+"");
    }
    private void ensureCapacity(int size) {
        // TODO Auto-generated method stub
        int len = elements.length;
        if(size > len){
            int newLen = (len*3)/2+1; //每次扩容方式
            elements = Arrays.copyOf(elements, newLen);
        }
    }
    public void pop() {
        Integer top = top();
        if(top != null){
            elements[size-1] = (Integer) null;
        }
        size--;
        minStack.pop();    
        min = minStack.peek();
    //    System.out.println(min+"");
    }
    public int top() {
        if(!empty()){
            if(size-1>=0)
                return elements[size-1];
        }
        return (Integer) null;
    }
    public boolean empty(){
        return size == 0;
    }
    public int min() {
        return min;
    }
}