你现在是一场采特殊赛制棒球比赛的记录员。这场比赛由若干回合组成,过去几回合的得分可能会影响以后几回合的得分。
比赛开始时,记录是空白的。你会得到一个记录操作的字符串列表 ops,其中 ops[i] 是你需要记录的第 i 项操作,ops 遵循下述规则:
整数 x - 表示本回合新获得分数 x
“+” - 表示本回合新获得的得分是前两次得分的总和。题目数据保证记录此操作时前面总是存在两个有效的分数。
“D” - 表示本回合新获得的得分是前一次得分的两倍。题目数据保证记录此操作时前面总是存在一个有效的分数。
“C” - 表示前一次得分无效,将其从记录中移除。题目数据保证记录此操作时前面总是存在一个有效的分数。
请你返回记录中所有得分的总和。
方法一:使用ArrayList
时间复杂度:O(N)
空间复杂度:O(N)
class Solution {
public int calPoints(String[] ops) {
List<Integer> list = new ArrayList<>();
for(int i = 0; i < ops.length; i++){
if(ops[i].equals("+")){
list.add(list.get(list.size() - 1) + list.get(list.size() - 2));
}
else if(ops[i].equals("D")){
list.add(list.get(list.size() - 1) * 2);
}
else if(ops[i].equals("C")){
list.remove(list.size() - 1);
}
else{
list.add(Integer.parseInt(ops[i]));
}
}
int sum = 0;
for(int i : list){
sum += i;
}
return sum;
}
}
方法二:栈
时间复杂度:O(N)
空间复杂度:O(N)
class Solution {
public int calPoints(String[] ops) {
Stack<Integer> stk = new Stack<>();
for(String str : ops){
if(str.equals("+")){
int top = stk.pop();
int newtop = stk.peek() + top;
stk.push(top);
stk.push(newtop);
}
else if(str.equals("D")){
int newtop = stk.peek() * 2;
stk.push(newtop);
}
else if(str.equals("C")){
stk.pop();
}
else{
stk.push(Integer.parseInt(str));
}
}
int sum = 0;
for(Integer i : stk){
sum += i;
}
return sum;
}
}