30 包含min函数的栈

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

原创最新推荐文章于 2023-07-24 10:28:56 发布 · 70 阅读

0 ·

CC 4.0 BY-SA版权

剑指offer 专栏收录该内容

57 篇文章

订阅专栏

本文介绍了一种使用两个栈的数据结构来实现在O(1)时间复杂度内获取栈中最小元素的方法。通过一个主栈存储所有元素，辅助栈则用于跟踪并保持栈中当前的最小值。当新元素被压入栈时，会与辅助栈的栈顶元素进行比较，确保辅助栈的栈顶始终是最小值。这种方法在不牺牲插入和删除操作效率的同时，提供了快速获取最小值的功能。

package sort;

import java.util.Stack;

public class Test30 {

public static Stack<Integer> stack = new Stack<Integer>(); // 用于存放所有数
public static Stack<Integer> storemin = new Stack<Integer>();// 用于存放当前栈的最小数实现O(1)取最小数

   public static void push(int s) {// 放元素时主栈直接放，辅助栈需要一些比较，比栈顶元素小再放，否则再次存放栈顶元素，
                                   // 保证辅助栈栈顶永远是最小的数
       stack.push(s);
       if (storemin.isEmpty()) {
           storemin.push(s);
       } else if (storemin.peek() > s) {
           storemin.push(s);
       } else if (storemin.peek() <= s) {
           storemin.push(storemin.peek());
       }

}

public static int pop() { // 需要同时取两个栈的栈顶元素，返回主栈的栈顶元素
if (!stack.isEmpty() && !storemin.isEmpty()) {

           storemin.pop();
           return stack.pop();
       }
       return 0;

}

public static int min() {// 直接去辅助栈的栈顶，复杂度为O(1);
return storemin.peek();

}

   public static void main(String[] args) {
       push(3);
       push(4);
       push(5);
       System.out.println(min());
       System.out.println(pop());
       System.out.println(min());
       System.out.println(pop());
       System.out.println(pop());
   }
}