剑指offer(15)

Posted 2021-02-11 figsprite

 

题目：

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

书中的思路：　　

　　按照这个思路我们很容易写出以下代码：

import java.util.Stack;

public class Solution {

    Stack stack = new Stack();
    Stack assistStack = new Stack();
    
    public void push(int node) {
        stack.push(node);
        if(assistStack.isEmpty()){
            assistStack.push(node);
        }else if((int)assistStack.peek()>node){
            assistStack.push(node);
        }else{
            assistStack.push((int)assistStack.peek());
        }
    }
    
    public void pop() {
        stack.pop();
        assistStack.pop();
    }
    
    public int top() {
        return (int)stack.peek();
    }
    
    public int min() {
        return (int)assistStack.peek();
    }
}

　　　　设置一个辅助栈，存放当前最小的值。