题解 | #最大体重的牛#
最大体重的牛
https://www.nowcoder.com/practice/0333d46aec0b4711baebfeb4725cb4de
考察栈的应用。需要利用两个栈结构,一个栈正常存储,另一个栈只存储当前另一个栈当前数据排列的最大值。配合着执行规定的操作就可以了,尤其是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; } }