LeetCode 682. Baseball Game

LeetCode 682. Baseball Game (棒球比赛)

题目

链接

https://leetcode-cn.com/problems/baseball-game/

问题描述

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

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

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

输入：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

示例

提示

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

思路

直接模拟，设置一个栈，如果需要操作之前的数，就从栈顶取数。

复杂度分析

时间复杂度 O(n)
空间复杂度 O(n)

代码

Java

    public int calPoints(String[] ops) {
        Stack<Integer> stack = new Stack<>();
        int sum = 0;
        for (String op : ops) {
            if (op.equals("+")) {
                int x = stack.pop();
                int y = stack.peek();
                sum += x + y;
                stack.push(x);
                stack.push(x + y);
            } else if (op.equals("C")) {
                sum -= stack.pop();
            } else if (op.equals("D")) {
                sum += stack.peek() * 2;
                stack.push(stack.peek() * 2);
            } else {
                int num = Integer.valueOf(op);
                stack.push(num);
                sum += num;
            }
        }
        return sum;
    }