[Leetcode] 0682. 棒球比赛

682. 棒球比赛

点击上方标题跳转至leetcode

题目描述

你现在是一场采用特殊赛制棒球比赛的记录员。这场比赛由若干回合组成，过去几回合的得分可能会影响以后几回合的得分。

比赛开始时，记录是空白的。你会得到一个记录操作的字符串列表 ops，其中 ops[i] 是你需要记录的第 i 项操作，ops 遵循下述规则：

1. 整数 x - 表示本回合新获得分数 x
2. "+" - 表示本回合新获得的得分是前两次得分的总和。题目数据保证记录此操作时前面总是存在两个有效的分数。
3. "D" - 表示本回合新获得的得分是前一次得分的两倍。题目数据保证记录此操作时前面总是存在一个有效的分数。
4. "C" - 表示前一次得分无效，将其从记录中移除。题目数据保证记录此操作时前面总是存在一个有效的分数。

请你返回记录中所有得分的总和。

示例 1：

输入：ops = ["5","2","C","D","+"]
输出：30
解释：
"5" - 记录加 5 ，记录现在是 [5]
"2" - 记录加 2 ，记录现在是 [5, 2]
"C" - 使前一次得分的记录无效并将其移除，记录现在是 [5].
"D" - 记录加 2 * 5 = 10 ，记录现在是 [5, 10].
"+" - 记录加 5 + 10 = 15 ，记录现在是 [5, 10, 15].
所有得分的总和 5 + 10 + 15 = 30

示例 2：

输入：ops = ["5","-2","4","C","D","9","+","+"]
输出：27
解释：
"5" - 记录加 5 ，记录现在是 [5]
"-2" - 记录加 -2 ，记录现在是 [5, -2]
"4" - 记录加 4 ，记录现在是 [5, -2, 4]
"C" - 使前一次得分的记录无效并将其移除，记录现在是 [5, -2]
"D" - 记录加 2 * -2 = -4 ，记录现在是 [5, -2, -4]
"9" - 记录加 9 ，记录现在是 [5, -2, -4, 9]
"+" - 记录加 -4 + 9 = 5 ，记录现在是 [5, -2, -4, 9, 5]
"+" - 记录加 9 + 5 = 14 ，记录现在是 [5, -2, -4, 9, 5, 14]
所有得分的总和 5 + -2 + -4 + 9 + 5 + 14 = 27

示例 3：

输入：ops = ["1"]
输出：1

提示：

• 1 <= ops.length <= 1000
• ops[i] 为 "C"、"D"、"+"，或者一个表示整数的字符串。整数范围是 [-3 * 104, 3 * 104]
• 对于 "+" 操作，题目数据保证记录此操作时前面总是存在两个有效的分数
• 对于 "C" 和 "D" 操作，题目数据保证记录此操作时前面总是存在一个有效的分数

解法

利用栈简单模拟即可。

Python3

from typing import List    

class Solution:
    def calPoints(self, operations: List[str]) -> int:
        sum = 0
        points = []
        for op in ops:
            if op =='+':
                pt = points[-1] + points[-2]
            elif op == 'D':
                pt = points[-1] *2
            elif op == 'C':
                sum -= points.pop()
                continue
            else:
                pt = int(op)
            sum += pt
            points.append(pt)
        return sum 
        
fun = Solution()
ops = ["5","2","C","D","+"]
res = fun.calPoints(ops)
print(res)

class Solution:
    def calPoints(self, ops: List[str]) -> int:
        stk = []
        for op in ops:
            if op == '+':
                stk.append(stk[-1] + stk[-2])
            elif op == 'D':
                stk.append(stk[-1] << 1)
            elif op == 'C':
                stk.pop()
            else:
                stk.append(int(op))
        return sum(stk)

C++

class Solution {
public:
    int calPoints(vector<string>& ops) {
        vector<int> stk;
        for (auto& op : ops) {
            int n = stk.size();
            if (op == "+") {
                int a = stk[n - 1];
                int b = stk[n - 2];
                stk.push_back(a + b);
            } else if (op == "D")
                stk.push_back(stk[n - 1] * 2);
            else if (op == "C")
                stk.pop_back();
            else
                stk.push_back(stoi(op));
        }
        return accumulate(stk.begin(), stk.end(), 0);
    }
};

#include<iostream>
#include<string>
#include<string>
#include<vector>
using namespace std;

class Solution {
public:
    int calPoints(vector<string>& operations) {
        int sum = 0;
        vector<int> points;
        for(auto &pt:operations){
            int n = points.size();
            switch(pt[0]){
                case '+':
                    sum += points[n-1] +points[n-2];
                    points.push_back(points[n-1] +points[n-2]);
                    break;
                case 'D':
                    sum += 2 * points[n-1];
                    points.push_back(2 * points[n-1]);
                    break;
                case 'C':
                    sum -= points[n-1];
                    points.pop_back();
                    break;
                default:
                    sum +=stoi(pt);
                    points.push_back(stoi(pt));
                    break;
            }
        }
        return sum;
    }
};


int main(){

    vector<string> ops = {"5","2","C","D","+"};
    // string s ="abc";
    int res = Solution().calPoints(ops);
    cout << res << endl;
    return 0;
}
//g++ 682.cpp -std=c++11