682. Baseball Game

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

Baseball Game - LeetCode

난이도 : easy

문제

총 합산점수를 반환하라

 

문자열이 + D C 그리고 숫자가 입력으로 들어온다 + 앞에 두수를 더하고 C 면 앞에수를 빼고 D 면 앞에수에 2를 곱한 값을 삽입한다. 

전형적인 구현문제로서 Stack 자료형으로 쉽게 풀 수 있다.

 

코드

class Solution {
public:
    int calPoints(vector<string>& ops) {
        
        stack<int> st;
        
        for(string c:ops) {
            
            if(c == "C") {
                st.pop();
            }
            else if(c == "D") {
                st.push(st.top() * 2);
            }
            else if(c == "+") {
                int top = st.top(); st.pop();
                int next = st.top(); 
                st.push(top);
                st.push(top + next);
            }
            else {
                st.push(stoi(c));
            }
        }
        
        int ans = 0;
        while(!st.empty()) {
            ans += st.top();
            st.pop();
        }
        
        return ans;
    }
};