20 包含min函数的栈

题目:包含min函数的堆栈

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

 

 

import java.util.Stack;
public class Solution {
    //借助于辅助栈,一个用来存放元素,一个用来存放最小值
    Stack element = new Stack();
    Stack minstack = new Stack();
    public void push(int node) {
        element.push(node);
        //如果最小栈为空或者是新输入的值比最小栈的值还要小,则将新数据入站
        //强制性转换容易忽略,如果去掉int会如下错误
        //./Solution.java:19: error: bad operand types for binary operator '>='
        //ifinstack.empty() || minstack.peek() >= node)
        //^
        //first type: Object
        //second type: int
        //Note: ./Solution.java uses unchecked or unsafe operations.
        //Note: Recompile with -Xlint:unchecked for details.
        //1 error
        if( minstack.empty() || (int)minstack.peek() >= node)
            minstack.push(node);
    }
    
    public void pop() {
        //元素栈正常出栈,但是出栈之前要检查,该元素是否与最小值相等,若相等,则最小值栈也要pop一个元素
        if(element.empty()) return;
        if(element.peek() == minstack.peek()){
            element.pop();
            minstack.pop();
        }else{
            element.pop();
        }
    }
    
    public int top() {
        return (int)element.peek(); //强制性转换容易忽略
    }
    
    public int min() {
        return (int)minstack.peek(); //强制性转换容易忽略
    }
}

 

 

 

本地编译器代码