682. 棒球比赛 简单

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

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

  • 整数 x - 表示本回合新获得分数 x

  • “+” - 表示本回合新获得的得分是前两次得分的总和。题目数据保证记录此操作时前面总是存在两个有效的分数。

  • “D” - 表示本回合新获得的得分是前一次得分的两倍。题目数据保证记录此操作时前面总是存在一个有效的分数。

  • “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” 操作,题目数据保证记录此操作时前面总是存在一个有效的分数

代码参考:

package main

import (
    "fmt"
    "strconv"
)

func main() {
    fmt.Println(calPoints([]string{"5", "2", "C", "D", "+"}))                 // 30
    fmt.Println(calPoints([]string{"5", "-2", "4", "C", "D", "9", "+", "+"})) // 27
}

// 有点后缀表达式计算值的感觉
func calPoints(ops []string) int {
    var s Stack
    for _, str := range ops {
        switch str {
        case "+":
            if !s.IsEmpty() {
                originPeek := s.Pop()
                sum := originPeek + s.Peek() // 所有 case 前两个元素都已是数字
                s.Push(originPeek)
                s.Push(sum)
            }
        case "D":
            if !s.IsEmpty() {
                s.Push(2 * s.Peek())
            }
        case "C":
            if !s.IsEmpty() {
                s.Pop()
            }
        default:
            n, _ := strconv.Atoi(str)
            s.Push(n)
        }
    }

    sum := 0
    for !s.IsEmpty() {
        sum += s.Pop()
    }

    return sum
}
最后编辑: kuteng  文档更新时间: 2021-06-05 10:16   作者:kuteng