20. 包含 min 函数的栈

原创于 2019-03-13 22:39:00 发布 · 162 阅读

0 ·

CC 4.0 BY-SA版权

本文介绍了一种在栈数据结构中高效获取最小元素的方法。通过使用两个栈，一个用于存储所有元素，另一个专门用于跟踪最小值，实现了在O(1)时间内获取当前栈中的最小元素。这种方法在不增加额外查询时间的情况下，保持了栈的基本操作。

题目描述

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

解题思路

直接用一个 minStack 栈维护最小元素就行了

代码实现

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

public void push(int node) {
    dataStack.push(node);
    minStack.push(minStack.isEmpty() ? node : Math.min(minStack.peek(), node));
}

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

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

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