实现一个可以返回其所存储的最小元素的栈

原创于 2016-08-01 21:02:09 发布 · 239 阅读

0 ·

CC 4.0 BY-SA版权

本文介绍了一种使用两个栈来实现在O(1)时间内获取栈中最小元素的方法。一个栈用于存储所有元素，另一个栈则用于存储当前栈中的最小值。通过这种方式，可以确保在任何时候都能快速获取到栈中的最小元素。

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

思路：用两个栈来存储元素，一个用来存储栈的所有元素，一个用来筛选栈的最小元素，利用栈先入后出的思路，只要入栈的元素小于筛选栈的栈顶元素就将这个元素也压入筛选栈，这样就能保证筛选栈的栈顶元素始终是栈中的最小元素，出栈同理，代码如下：

import java.util.Stack;

public class Solution {

    private Stack<Integer> stackData;
    private Stack<Integer> stackMin;
    
    public Solution() {
        stackData = new Stack<>();
        stackMin = new Stack<>();
    }
    
    public void push(int node) {
        if(stackMin.empty()){
            stackMin.push(node);
        }else if(node<this.min()){
            stackMin.push(node);
        }
        stackData.push(node);
    }
    
    public void pop() {
        int value = stackData.pop();
        if(value == this.min()){
            stackMin.pop();
        }
    }
    
    public int min() {
        return stackMin.peek();
    }
}