682. 棒球比赛（栈）

package com.heu.wsq.leetcode.mystack.common;

import java.util.ArrayDeque;
import java.util.Deque;

/**
 * 682. 棒球比赛
 * @author wsq
 * @date 2020/12/27
 * 你现在是一场采特殊赛制棒球比赛的记录员。这场比赛由若干回合组成，过去几回合的得分可能会影响以后几回合的得分。
 * 比赛开始时，记录是空白的。你会得到一个记录操作的字符串列表 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
 *
 * 链接：https://leetcode-cn.com/problems/baseball-game
 */
public class CalPoints {
    /**
     *
     * @param ops
     * @return
     */
    public int calPoints(String[] ops){
        int n = ops.length;

        if (n == 0){
            return 0;
        }
        // 存储当前分数
        Deque<Integer> stack = new ArrayDeque<>();
        for (int i = 0; i < n; i++){
            if (ops[i].equals("C")){
                stack.removeLast();
            }else if (ops[i].equals("D")){
                stack.addLast(stack.peekLast() * 2);
            }else if (ops[i].equals("+")){
                Integer tmp = stack.removeLast();
                int addN = tmp + stack.peekLast();
                // 入栈
                stack.addLast(tmp);
                stack.addLast(addN);
            }else{
                stack.addLast(Integer.parseInt(ops[i]));
            }
        }
        int sum = 0;
        while (!stack.isEmpty()){
            sum += stack.removeLast();
        }
        return sum;
    }

    public static void main(String[] args) {
        String[] ops = {"5","2","C","D","+"};
        CalPoints cp = new CalPoints();
        int ans = cp.calPoints(ops);
        System.out.println(ans);
    }
}