包含min函数的栈JAVA实现

最新推荐文章于 2022-07-04 20:19:07 发布

原创最新推荐文章于 2022-07-04 20:19:07 发布 · 2.1k 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#最小栈

数据结构与算法专栏收录该内容

79 篇文章

订阅专栏

本文介绍了一种特殊栈的设计方案，该栈除了具备常规的push和pop操作外，还额外提供了一个min函数用于获取栈中当前的最小值。通过使用两个栈结构，一个用于存储数据元素，另一个专门用来维护最小值，从而确保了所有操作的时间复杂度均为O(1)。



/*
 * 设计包含min 函数的栈。
 * 定义栈的数据结构，要求添加一个min 函数，能够得到栈的最小元素。
 * 要求函数min、push 以及pop 的时间复杂度都是O(1)。
 * 思路：设计栈的元素同时包含当前值和当前最小值，这样当pop掉最小值的时候，栈顶元素依然保存着当前最小值

 * 或者直接用两个栈，一个存数据，一个存最小值
 * */

import java.util.Stack;


public class Solution {
    Stack<Integer> stack_data=new Stack<>();
    Stack<Integer> stack_min=new Stack<>();
    public void push(int node) {
       stack_data.push(node); 
       if (stack_min.isEmpty()) {
    	   stack_min.push(node);
       }else if(node<min()){
           stack_min.push(node);
       }else {
    	   stack_min.push(min());
       }
    }
    
    public void pop() {
        stack_data.pop();
        stack_min.pop();
    }
    
    public int top() {
        return stack_data.peek();
    }
    
    public int min() {
        return stack_min.peek();
    }
}