题解 | #最大体重的牛# java
最大体重的牛
https://www.nowcoder.com/practice/0333d46aec0b4711baebfeb4725cb4de
import java.util.*; public class Solution { /** * 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可 * * * @param op string字符串一维数组 * @param vals int整型二维数组 * @return int整型一维数组 */ Stack<Integer> s = new Stack<>(); Stack<Integer> maxs = new Stack<>(); public int[] max_weight_cow(String[] op, int[][] vals) { int[] res = new int[op.length]; for (int i = 0; i < op.length; i++) { if (op[i].equals("MaxCowStack")) { res[i] = -1; } else if (op[i].equals("push")) { s.push(vals[i][1]); if (maxs.isEmpty() || vals[i][1] > maxs.peek()) { maxs.push(vals[i][1]); } res[i] = -1; } else if (op[i].equals("pop")) { int val = s.pop(); if (maxs.peek() == val) { maxs.pop(); } res[i] = -1; } else if (op[i].equals("top")) { res[i] = s.peek(); } else if (op[i].equals("getMax")) { res[i] = maxs.peek(); } } return res; } }
Java语言。
此题考察的知识点是栈的应用。
以下是对代码的文字解释:
使用循环遍历 op 数组,并根据不同的操作类型执行相应的操作,并将结果存入 res 数组中。如果当前操作是 "MaxCowStack",则将 res 数组对应位置赋值为 -1。
如果当前操作是 "push",则将 vals 数组中指定位置的元素入栈,并判断是否需要更新最大值栈。如果当前操作是 "pop",则将栈顶元素弹出,并判断是否需要更新最大值栈。
如果当前操作是 "top",则将栈顶元素存入 res 数组对应位置。如果当前操作是 "getMax",则将最大值栈顶元素存入 res 数组对应位置。
最后,返回结果数组 res。