剑指offer系列-----item20 带min函数的stack栈

最新推荐文章于 2022-01-24 13:03:51 发布

Sirius_7

最新推荐文章于 2022-01-24 13:03:51 发布

阅读量103

点赞数

分类专栏：算法 JAVA 数据结构文章标签：数据结构算法栈 stack java

本文链接：https://blog.csdn.net/weixin_38107316/article/details/105792701

版权

JAVA 同时被 3 个专栏收录

76 篇文章 0 订阅

订阅专栏

算法

61 篇文章 0 订阅

订阅专栏

数据结构

50 篇文章 0 订阅

订阅专栏

本文介绍了一种使用两个栈实现数据结构的方法，其中一个栈用于正常存储数据，另一个栈专门用于跟踪并存储当前栈中的最小值。这种方法确保了在O(1)时间内获取栈中的最小元素，避免了传统排序方法的高时间复杂度问题。

摘要由CSDN通过智能技术生成

题干：

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

思路：

思路，一开始想着是压根不用栈，直接利用集合模拟栈这种数据结构的pop、push、top，最后再利用Collections工具类进行sort排序，输出第一个值即可。不过这样存在一个问题，时间复杂度太高了，Collection.sort的时间复杂度在O(n2)以上。那么改变思路，用两个栈来实现该算法功能：
1.两个栈一个正常存放数据，另一个存放最小值；
2.存放最小值的栈内数据的数量和另一个栈要保持一致；
3.存放最小值的栈的数据更新策略为：
（1）为空时直接和另一个栈添加一样的数据；
（2）不为空时，如果node比栈顶元素小，那么添加node，反之添加原栈顶元素。

具体解法如下：

import java.util.Stack;
public class Solution {
    Stack<Integer> stack = new Stack<>();
    Stack<Integer> stackmin = new Stack<>();
    public void push(int node) {
        stack.push(node);
        if(stackmin.empty()){
            stackmin.push(node);
        }
        else {
            if(stackmin.peek()>node){
                stackmin.push(node);
            }
            else {
                stackmin.push(stackmin.peek());
            }
        }
    }
    public void pop() {
        stack.pop();
        stackmin.pop();
    }
    public int top() {
        return stack.peek();
    }
    public int min() {
        return stackmin.peek();
    }
}