题解 | #最大体重的牛#

题目考察的知识点: 栈

题目解答方法的文字分析:

stack用于存储元素的栈，maxStack用于存储每个元素入栈时，这个时刻的最大值。
push时，第一个普通栈直接压栈，第二个栈maxStack需要判断当前元素到栈底的最大值，将最大值压栈。此时maxStack中栈顶元素就是所有牛的最大重量。
pop时，对两个栈都出栈

本题解析所用的编程语言:Java

完整且正确的编程代码

import java.util.*;


public class Solution {
    /**
     * 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
     *
     *
     * @param op string字符串一维数组
     * @param vals int整型二维数组
     * @return int整型一维数组
     */
    public int[] max_weight_cow (String[] op, int[][] vals) {
        // write code here
        Deque<Integer> stack = new ArrayDeque<>();
        Deque<Integer> maxStack = new ArrayDeque<>();
        int n = op.length;
        int[] ans = new int[n];
        for (int i = 0; i < n; i++) {
            ans[i] = -1;
            if (op[i].equals("MaxCowStack")) {
                stack = new ArrayDeque<>();
                maxStack = new ArrayDeque<>();
            } else if (op[i].equals("push")) {
                stack.push(vals[i][1]);
                if (!maxStack.isEmpty() && maxStack.peek() > vals[i][1]) {
                    maxStack.push(maxStack.peek());
                } else {
                    maxStack.push(vals[i][1]);
                }
            } else if (op[i].equals("pop")) {
                stack.pop();
                maxStack.pop();
            } else if (op[i].equals("top")) {
                ans[i] = stack.peek();
            } else {
                ans[i] = maxStack.peek();
            }
        }
        return ans;
    }
}