查看题目详情可点击此处。
题目
你现在是棒球比赛记录员。
给定一个字符串列表,每个字符串可以是以下四种类型之一:
- 整数(一轮的得分):直接表示您在本轮中获得的积分数。
- "+"(一轮的得分):表示本轮获得的得分是前两轮有效 回合得分的总和。
- "D"(一轮的得分):表示本轮获得的得分是前一轮有效 回合得分的两倍。
- "C"(一个操作,这不是一个回合的分数):表示您获得的最后一个有效 回合的分数是无效的,应该被移除。
每一轮的操作都是永久性的,可能会对前一轮和后一轮产生影响。
你需要返回你在所有回合中得分的总和。
示例 1:
输入: ["5","2","C","D","+"]
输出: 30
解释:
第1轮:你可以得到5分。总和是:5。
第2轮:你可以得到2分。总和是:7。
操作1:第2轮的数据无效。总和是:5。
第3轮:你可以得到10分(第2轮的数据已被删除)。总数是:15。
第4轮:你可以得到5 + 10 = 15分。总数是:30。
示例 2:
输入: ["5","-2","4","C","D","9","+","+"]
输出: 27
解释:
第1轮:你可以得到5分。总和是:5。
第2轮:你可以得到-2分。总数是:3。
第3轮:你可以得到4分。总和是:7。
操作1:第3轮的数据无效。总数是:3。
第4轮:你可以得到-4分(第三轮的数据已被删除)。总和是:-1。
第5轮:你可以得到9分。总数是:8。
第6轮:你可以得到-4 + 9 = 5分。总数是13。
第7轮:你可以得到9 + 5 = 14分。总数是27。
注意:
输入列表的大小将介于1和1000之间。
列表中的每个整数都将介于-30000和30000之间。
解题思路
因为计算涉及到的数据最多为最近两轮的数据,适合使用栈解决,C 操作就是出栈,整数就是入栈当前局得分,D 操作就是以栈顶数据的双倍入栈,+ 操作就是以栈顶数据及栈顶的下一个数据之各入栈,最后再将栈中的数据加和就可解题。
代码实现
class Solution {
public int calPoints(String[] ops) {
Stack<Integer> stack = new Stack<>();
for(String op : ops) {
if("+".equals(op)) {
Integer top = stack.pop();
Integer newTop = stack.peek() + top;
stack.push(top);
stack.push(newTop);
} else if("D".equals(op)) {
stack.push(stack.peek() * 2);
} else if("C".equals(op)) {
stack.pop();
} else {
stack.push(Integer.parseInt(op));
}
}
Integer result = 0;
for(Integer val : stack) {
result += val;
}
return result;
}
}