欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

剑指offer30.包含min函数的栈

程序员文章站 2022-07-10 13:58:09
...

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

因为时间复杂度要求维O(1),就需要直接读取到该栈中的最小值,考虑到在一个栈上不可能实现随机存取,就需要考虑使用辅助栈记录来完成。


public class Solution {

    

        Stack<Integer> stack1 = new Stack<>();
        Stack<Integer> stack2 = new Stack<>();

        public void push(int node) {
            stack1.push(node);
            if (stack2.isEmpty()){
                stack2.push(node);
            }else {
                if (stack2.peek()<node){
                    stack2.push(stack2.peek());
                }else {
                    stack2.push(node);
                }
            }

        }

        public void pop() {
            stack1.pop();
            stack2.pop();

        }

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

        public int min() {
           return stack2.peek();

        }
}