LeetCode 682 棒球比赛[栈 模拟] HERODING的LeetCode之路

Posted 2022-03-28 HERODING23

解题思路：
阅读完题目就能明白这是最简单的栈操作处理，直接初始化一个栈，按照题目的要求进行操作即可，代码如下：

class Solution 
public:
    int calPoints(vector<string>& ops) 
        stack<int> s;
        for(int i = 0; i < ops.size(); i ++) 
            if(ops[i] == "+") 
                int first = s.top();
                s.pop();
                int second = s.top();
                s.push(first);
                s.push(first + second);
             else if(ops[i] == "D") 
                s.push(2 * s.top());
             else if(ops[i] == "C") 
                s.pop();
             else 
                s.push(stoi(ops[i]));
            
        
        int ans = 0;
        while(!s.empty()) 
            ans += s.top();
            s.pop();
        
        return ans;
    
;

但是仔细观察你就能发现，其实这其中最多只涉及栈最后两个元素的操作，那不如定义成数组，模拟栈的操作，入栈的过程中统计栈中元素和，思路也是同样的，代码如下：

class Solution 
public:
    int calPoints(vector<string>& ops) 
        vector<int> s;
        int ans = 0;
        for(int i = 0; i < ops.size(); i ++) 
            int n = s.size();
            if(ops[i] == "+") 
                ans += s[n - 1] + s[n - 2];
                s.push_back(s[n - 1] + s[n - 2]);
             else if(ops[i] == "D") 
                ans += 2 * s[n - 1];
                s.push_back(2 * s[n - 1]);
             else if(ops[i] == "C") 
                ans -= s[n - 1];
                s.pop_back();
             else 
                ans += stoi(ops[i]);
                s.push_back(stoi(ops[i]));
            
        
        return ans;
    
;