Leetcode No.682 棒球比赛

题目大意

你现在是棒球比赛记录员。
给定一个字符串列表，每个字符串可以是以下四种类型之一：
1.整数（一轮的得分）：直接表示您在本轮中获得的积分数。

"+"（一轮的得分）：表示本轮获得的得分是前两轮有效回合得分的总和。
"D"（一轮的得分）：表示本轮获得的得分是前一轮有效回合得分的两倍。
"C"（一个操作，这不是一个回合的分数）：表示您获得的最后一个有效回合的分数是无效的，应该被移除。

每一轮的操作都是永久性的，可能会对前一轮和后一轮产生影响。
你需要返回你在所有回合中得分的总和。
示例1：

输入: ["5","2","C","D","+"]
输出: 30
解释:
第1轮：你可以得到5分。总和是：5。
第2轮：你可以得到2分。总和是：7。
操作1：第2轮的数据无效。总和是：5。
第3轮：你可以得到10分（第2轮的数据已被删除）。总数是：15。
第4轮：你可以得到5 + 10 = 15分。总数是：30。

示例2：

输入: ["5","-2","4","C","D","9","+","+"]
输出: 27
解释:
第1轮：你可以得到5分。总和是：5。
第2轮：你可以得到-2分。总数是：3。
第3轮：你可以得到4分。总和是：7。
操作1：第3轮的数据无效。总数是：3。
第4轮：你可以得到-4分（第三轮的数据已被删除）。总和是：-1。
第5轮：你可以得到9分。总数是：8。
第6轮：你可以得到-4 + 9 = 5分。总数是13。
第7轮：你可以得到9 + 5 = 14分。总数是27。

注意：
输入列表的大小将介于1和1000之间。
列表中的每个整数都将介于-30000和30000之间。

题目分析

用栈来存储每一轮的得分，用一个变量sum动态更新得分总和。一共有四种情况：

遇到数字，直接入栈，并且将该数值加入sum中。
遇到"C"，表示上轮分数无效，弹出栈顶元素v,并将sum减去v。
遇到"D"，表示本轮得分是上轮有效得分的双倍。获取栈顶元素v，本轮得分为2v，将本轮得分入栈，并且sum = sum+2v。
遇到"+", 获得栈的top2元素v1和v2，本轮得分v = v1+v2。将v入栈，然后sum += v。

代码一：栈

public int calPoints(String[] ops) {
        if(ops == null || ops.length == 0) return 0;
        
        int sum = 0;
        Stack<Integer> stack = new Stack<>();

        for(int i=0;i<ops.length;i++) {
            if(ops[i].equals("C")) {
                int v = stack.pop();  //取无效的分数
                sum -= v;  //更新sum
            } else if(ops[i].equals("D")) {
                int v = stack.peek();
                stack.push(v*2);
                sum += v*2;
            } else if(ops[i].equals("+")){   // +
                int last1 = stack.pop();
                int last2 = stack.peek();
                stack.push(last1);
                int v = last1 + last2;
                stack.push(v);
                sum += v;
            } else{
                int val = Integer.valueOf(ops[i]);
                stack.push(val);
                sum += val; //更新sum
            }
        }
        return sum;
    }

注意字符串的内容比较，用equals函数。刚开始我直接用了ops[i] == "C"，会报错。运行结果3ms,击败85.76%。

代码二：优化版本

这是看网上别人写的代码，用int数组代替了stack。

public int calPoints(String[] ops) {
        if(ops == null || ops.length == 0) return 0;
        int[] arr = new int[ops.length];
        int sum = 0;
        int i=0;
        for(String s:ops) {
            switch(s) {
                case "+":arr[i] = arr[i-1]+arr[i-2];i++;break;
                case "D":arr[i] = 2*arr[i-1];i++;break;
                case "C":arr[i-1] = 0;i--;break;
                default:
                    arr[i] = Integer.valueOf(s);
                    i++;
            }
        }
        for(int v:arr) sum += v;
        return sum;
    }

运行时间2ms,击败93.94%。