剑指offer

涉及的知识点

栈

题目描述

定义栈的数据结构，请在该类型中实现一个能够得到栈的最小元素的 min 函数。在该栈中，调用 min、push 及 pop 函数的时间复杂度都是 O(1).
push(value)：将元素 value压入栈中
pop()：弹出栈顶的元素
top() ：获得栈顶元素
min() ：获得栈中的最小元素。

示例:
输入:
[“PSH-1”,“PSH2”,“MIN”,“TOP”,“POP”,“PSH1”,“TOP”,“MIN”]
输出:
-1,2,1,-1

解释如下:
“PSH-1”: 将元素-1压入栈中，栈中元素为-1
”PSH2": 将元素2压入栈中，栈中元素为2, -1
“MIN”： 获取最小的元素 ==>输出-1
“TOP”: 获得栈顶元素 ==>输出2
“POP”: 弹出栈顶元素，即弹出2，栈中元素为-1
“PSH1”： 将元素1压入栈中，栈中元素为1，-1
“TOP”: 获得栈顶元素 ==>输出1
“MIN”： 获取最小的元素 ==>输出-1

解题思路

用两个栈，一个用来正常操作，一个用来存最小值

代码

import java.util.Stack;

public class Solution {

    Stack<Integer> normalStack = new Stack<>();
    Stack<Integer> minStack = new Stack<>();
    public void push(int node) {
        normalStack.push(node);
        if(minStack.isEmpty()){
             minStack.push(node);
        }else{
            if(node<=minStack.peek()){
                minStack.push(node);
            }else{
                minStack.push(minStack.peek());
            }
        }
      
    }
    
    public void pop() {
       normalStack.pop();
       minStack.pop();
    }
    
    public int top() {
        return normalStack.peek();
    }
    
    public int min() {
        return minStack.peek();
    }
}

复杂度

时间复杂度：O(n)
空间复杂度：O(n)