数据结构--java语言实现最小栈

目录

1 题目描述

2 解题思路

3 代码实现

---

1 题目描述

设计一个支持 push，pop，top 操作，并能在常数时间内检索到最小元素的栈。

• push(x) -- 将元素 x 推入栈中。
• pop() -- 删除栈顶的元素。
• top() -- 获取栈顶元素。
• getMin() -- 检索栈中的最小元素。

2 解题思路

利用两个栈集合stack和minStack：stack栈中存放入栈元素，minStack栈中存放最小元素，minStack入栈操作时判断stack栈栈顶元素和将要入栈元素的大小，若将要入栈的元素小于stack栈栈顶的元素，才进行入栈操作。这样得到的minStack栈就是最小栈啦~

3 代码实现

class MinStack {

    /** initialize your data structure here. */
    Stack<Integer> stack = new Stack<>();
    Stack<Integer> minStack = new Stack<>();
    public MinStack() {

    }
    //入栈时让stack栈入元素，minStack栈判断栈顶元素和将要入栈元素的大小，
    //当将要入栈的元素小于栈顶元素时才进行入栈操作
    public void push(int x) {
        stack.push(x);
        if (minStack.empty()){
            minStack.push(x);
        }else {
            if (minStack.peek()>=x){
                minStack.push(x);
            }
        }
    }
    //出栈时若两个栈顶元素相同，则将两个都进行出栈，否则只将stack栈的栈顶元素进行出栈
    public void pop() {
        if (!stack.empty()){
            int tmp = stack.pop();
            if(tmp == minStack.peek()){
                minStack.pop();
            }
        }
    }

    public int top() {
        if(stack.empty()){
            return -1;
        }
        return stack.peek();
    }

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