LeetCode 155.最小栈

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

实现 MinStack 类:

MinStack() 初始化堆栈对象。

void push(int val) 将元素val推入堆栈。

void pop() 删除堆栈顶部的元素。

int top() 获取堆栈顶部的元素。

int getMin() 获取堆栈中的最小元素。

示例 1:

输入:

["MinStack","push","push","push","getMin","pop","top","getMin"]

[[],[-2],[0],[-3],[],[],[],[]]

输出:

[null,null,null,null,-3,null,0,-2]

解释:

MinStack minStack = new MinStack();

minStack.push(-2);

minStack.push(0);

minStack.push(-3);

minStack.getMin(); --> 返回 -3.

minStack.pop();

minStack.top(); --> 返回 0.

minStack.getMin(); --> 返回 -2.

提示:

1、-231 <= val <= 231 - 1

2、pop、top 和 getMin 操作总是在 非空栈 上调用

3、push, pop, top, and getMin最多被调用 3 * 104 次

思路:

建立一个正常栈,另外一个栈为最小栈

  1. push方法:如果第二个元素大于第一个元素,则最小栈不入,正常栈入,反之,都入

  1. pop方法:正常栈出,直到出的元素等于最小栈的栈顶元素,都出

代码:

class MinStack {

    private Stack stack;
    private Stack minStack;

    public MinStack() {
        this.stack=new Stack<>();
        this.minStack=new Stack<>();
    }
    
    public void push(int val) {
        stack.push(val);
        if (minStack.empty()){
            minStack.push(val);
        }else {
            if (val<=minStack.peek()) {
                minStack.push(val);
            }
        }
    }
    
    public void pop() {
        if (stack.empty()){
            return;
        }
        int x=stack.pop();
        if (x==minStack.peek()){
            minStack.pop();
        }
    }
    
    public int top() {
        if (stack.empty()){
            return -1;
        }
        return stack.peek();
    }
    
    public int getMin() {
        if (minStack.empty()){
            return -1;
        }
        return minStack.peek();
    }
}

你可能感兴趣的:(数据结构,java,leetcode)