栈-包含min函数的栈

letcode原题:定义栈的数据结构,请在该类型中实现一个能够得到栈的最小元素的 min 函数在该栈中,调用 min、push 及 pop 的时间复杂度都是 O(1)。

示例:

MinStack minStack = new MinStack();
minStack.push(-2);
minStack.push(0);
minStack.push(-3);
minStack.min();   --> 返回 -3.
minStack.pop();
minStack.top();      --> 返回 0.
minStack.min();   --> 返回 -2.

提示:

各函数的调用总次数不超过 20000 次,top函数就是官方的peek()函数

来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/bao-han-minhan-shu-de-zhan-lcof
解题思路

1:该题的难点在于如何确保每次调用min函数都能返回当前栈中的最小元素值,因此我们需要记录这个最小值,如果用变量记录,那么只能记录入栈的全部元素的最小值,不能保证当前栈中的最小元素值的时效性,因此可以采取辅助栈的方式,我们可以使用一个StackIn来保存每次入栈的元素,采取一个minStack来记录每次入栈时的当前最小元素值,如下列代码.

class MinStack {

    private Stack<Integer> stack;
    private Stack<Integer> minStack;
    /** initialize your data structure here. */
    public MinStack() {
        this.stack=new Stack<>();
        this.minStack=new Stack<>();
    }

    public void push(int x) {
        stack.push(x);
        if(minStack.empty())
            minStack.push(x);
        else {
            if(x<=minStack.peek())
                minStack.push(x);
            else minStack.push(minStack.peek());//每个元素入栈时,minStack都会存入一个当前栈中最小元素
        }
    }

    public void pop() {
        minStack.pop();
        stack.pop();
    }

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

    public int min() {
        return minStack.peek();
    }
}

2:采用链表的形式来模拟栈,每个节点都记录当前链中的最小元素值

class MinStack {
    Node head;
    public MinStack() {
    }

    public void push(int x) {
        if(head==null)
            head=new Node(x,x,null);
        else{
            head=new Node(Math.min(x, head.min),x,head);
        }
    }
    public void pop() {
        head=head.next;
    }

    public int top() {
       return head.val;
    }

    public int min() {
      return head.min;
    }
  class Node{
        int min;
        int val;
        Node next;
        public Node(int min,int val,Node next){
            this.min=min;
            this.val=val;
            this.next=next;
        }
  }
}

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值