牛客刷题---JZ30 包含min函数的栈

JZ30 包含min函数的栈

---

描述

定义栈的数据结构，请在该类型中实现一个能够得到栈中所含最小元素的 min 函数，输入操作时保证 pop、top 和 min 函数操作时，栈中一定有元素。

此栈包含的方法有：

• push(value):将value压入栈中
• pop():弹出栈顶元素
• top():获取栈顶元素
• min():获取栈中最小元素

数据范围：

• 操作数量满足 0≤n≤300 ，
• 输入的元素满足 ∣val∣≤10000

进阶：栈的各个操作的时间复杂度是 O(1) ，空间复杂度是 O(n)

示例:
输入: [“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>  stack = new Stack<Integer> ();
    // 辅助栈，用于存当前栈最小值
    Stack<Integer>  minStack = new Stack<Integer> ();


    public void push(int node) {
        // 元素入栈
        stack.push(node);
        if (minStack.empty()) {
          // 如果辅助栈为空，node入栈
            minStack.push(node);
        } else {
            if (node < minStack.peek()) {
                minStack.push(node);
            } else {
                minStack.push(minStack.peek());
            }
        }
    }

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

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

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

复杂度分析

• 时间复杂度：O(N)
• 空间复杂度：O(1)