包含min函数的栈（栈）

最新推荐文章于 2021-10-10 22:35:49 发布

良殿世无双

最新推荐文章于 2021-10-10 22:35:49 发布

阅读量72

点赞数

分类专栏：剑指offer java 文章标签：栈辅助栈最小值数据结构 O(1)时间复杂度

本文链接：https://blog.csdn.net/codepandacode/article/details/115081529

版权

剑指offer 同时被 2 个专栏收录

45 篇文章 0 订阅

订阅专栏

java

16 篇文章 0 订阅

订阅专栏

题目

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

思路

辅助栈，设置数据栈和最小值栈，数据栈按普通栈的方式进行出栈入栈，最小值栈依次存入当前的最小值，栈顶元素永远是当前最小值。

代码

import java.util.Stack;

public class Solution {

    Stack <Integer> data = new Stack <>();
    Stack <Integer> mindata = new Stack <>();
   
    public void push(int node) {
        data.push(node);
       // if(!mindata.empty() && node < mindata.peek()){
        if(mindata.empty() ){
            mindata.push(node);
        }else{
            if(node < mindata.peek()){
                //mindata.push(mindata.peek());
                mindata.push(node);
            }else{
                mindata.push(mindata.peek());
            }
        }
//         if(!mindata.empty() && node <= mindata.peek()){
//             mindata.push(node);
//         }else{
//             mindata.push(mindata.peek());
//         }
    }
    
    public void pop() {
        if(!data.empty() && !mindata.empty()){
           // if(data.peek() == mindata.peek()){
                data.pop();
                mindata.pop();
          // }
        }
    }
    
    public int top() {
        return data.peek();
    }
    
    public int min() {
        return mindata.peek();
    }
}