题解 | #最大体重的牛#

知识点

栈，动态维护

解题思路

用一个栈来保存每次push的数据，用一个maxStack来动态维护当前最大值。

如果当前添加的数据比maxStack栈顶的元素大，那么添加进当前添加的值，否则将maxStack栈顶元素再次添加进maxStack。

这就和使用动态规划维护一个最大值数组一个道理，当前最大值=上一个元素最大值和当前元素中的大值。

arr[i] = max(arr[i - 1],num)。

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
        int[] ans = new int[op.length];
        MaxCowStack maxCowStack = new MaxCowStack();
        for (int i = 0; i < op.length; i++) {
            if (op[i].equals("MaxCowStack")) {
                ans[i] = -1;
            } else if (op[i].equals("push")) {
                ans[i] = maxCowStack.push(vals[i][1]);
            } else if (op[i].equals("pop")) {
                ans[i] = maxCowStack.pop();
            } else if (op[i].equals("top")) {
                ans[i] = maxCowStack.top();
            } else if (op[i].equals("getMax")) {
                ans[i] = maxCowStack.getMax();
            }
        }
        return ans;
    }

    class MaxCowStack {
        Stack<Integer> stack = new Stack<>();
        Stack<Integer> maxStack = new Stack<>();

        int push(int weight) {
            stack.push(weight);
            if (!maxStack.isEmpty() && maxStack.peek() > weight) {
                maxStack.push(maxStack.peek());
            } else {
                maxStack.push(weight);
            }
            return -1;
        }
        int pop() {
            stack.pop();
            maxStack.pop();
            return -1;
        }
        int top() {
            return stack.peek();
        }
        int getMax() {
            return maxStack.peek();
        }
    }
}