题解 | 包含min函数的栈

包含min函数的栈

https://www.nowcoder.com/practice/4c776177d2c04c2494f2555c9fcc1e49

import java.util.*;
import java.util.Stack;

public class Solution {

    Stack<Integer> s1 = new Stack<>();
    Stack<Integer> s2 = new Stack<>();

    public void push(int node) {
        s1.push(node);
	  // 这里的重点是s2.peek() >= node,需要将等于当前最小值的值也加入到s2中,这样可以确保与s1中与最小值相等的元素数量一致。
        if(s2.isEmpty() || s2.peek() >= node){
            s2.push(node);
        }
    }

    public void pop() {
        int num = s1.pop();
        if(num == s2.peek()){
            s2.pop();
        }
    }

    public int top() {
        return s1.peek();
    }

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

全部评论

相关推荐

评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客企业服务