JZ20 --- 包含min函数的栈

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

题解:
此题需要一个辅助栈,用来存当前最小值。

题解一

插入:

  1. 将结点入栈。
  2. 如果当前辅助栈为空,则插入结点。
  3. 如果插入结点小于辅助栈栈顶元素,则将当前结点插入辅助栈。

删除:

  1. 删除栈顶元素。
  2. 如果栈顶元素等于辅助栈栈顶元素,则辅助栈栈顶元素也需删除。

栈顶元素:

  1. 返回普通栈栈顶元素。

栈中最小元素:

  1. 返回辅助栈栈顶元素。
	ArrayList<Integer> stack = new ArrayList<> ();
	// 辅助栈
    ArrayList<Integer> minstack = new ArrayList<> ();
 
    public void push(int node) {
        stack.add (node);
        if(minstack.isEmpty ()){
            minstack.add (node);
        }
        if(node < minstack.get(minstack.size ()-1)){
            minstack.add (node);
        }
    }
 
    public void pop() {
        int val = stack.remove (stack.size () - 1);
        if(val == minstack.get (minstack.size () - 1)){
            minstack.remove (minstack.size () - 1);
        }
    }
 
    // 求栈顶元素
    public int top() {
        return stack.get (stack.size ()-1);
    }
 
    // 求栈中的最小元素
    public int min() {
        return minstack.get (minstack.size ()-1);
    }

题解二

插入:

  1. 将结点入栈。
  2. 如果当前辅助栈为空,则插入结点。
  3. 如果插入结点小于等于辅助栈栈顶元素,则将当前结点插入辅助栈。
  4. 反之将辅助栈栈顶元素再次插入辅助栈。

删除:

  1. 删除普通栈栈顶元素。
  2. 删除辅助栈栈顶元素。
	public void push(int node) {
        stack.add (node);
        if(minstack.isEmpty ()){
            minstack.add (node);
        }else if(node <= minstack.get (minstack.size ()-1)){
            minstack.add (node);
        }else{
            minstack.add (minstack.get (minstack.size ()-1));
        }
    }

    public void pop() {
        stack.remove (stack.size ()-1);
        minstack.remove (minstack.size ()-1);
    }

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值