剑指offer(java实现)第20题“包含min函数的栈”-牛客网

题目描述

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

时间限制：1秒 空间限制：32768K 热度指数：170246

解答：

import java.util.Stack;

public class Solution {//思路：用另一个栈stack2做辅助，当push一个元素到stack1中时，如果在push之前stack1的元素是空的，则用一个变量min记录下此时的元素，同时在stack2中也push进去
    //，如果stack1元素不为0，则先比较当前待push进去的元素是否小于min，如果是的，则用min指向该元素，同时stack2就push该元素，如果不小于min，则stack2就push min元素进去；这样能
    //确保stack2的顶层永远是最小元素

    private Stack<Integer> stack1 = new Stack();//保存所有元素的栈
    private Stack<Integer> stack2 = new Stack();//辅助栈，确保栈顶元素总是最小
    private int min = 0;
    public void push(int node) {
        if (stack1.empty()) {
            min = node;
        }else {
            if (min > node) {//这里其实可以放到上一个if语句中，但是为了看着更加直观就用一个新的if
                min = node;
            }
        }
        stack2.push(min);
        stack1.push(node);
    }
    
    public void pop() {//删除最顶层的
        stack1.pop();
        stack2.pop();
    }
    
    public int top() {//拿最顶层的元素，不删除
        return stack1.peek();//top()对应到java中就是peek()
    }
    
    public int min() {
        return stack2.peek();
    }
}