题解 | #最大体重的牛#
最大体重的牛
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;
}
}