[프로그래머스] 컨트롤 제트 (C++)

컨트롤 제트 / Lv.0

---

문제  설명 )

숫자와 "Z"가 공백으로 구분되어 담긴 문자열이 주어집니다. 문자열에 있는 숫자를 차례대로 더하려고 합니다. 이 때 "Z"가 나오면 바로 전에 더했던 숫자를 뺀다는 뜻입니다. 숫자와 "Z"로 이루어진 문자열 s가 주어질 때, 머쓱이가 구한 값을 return 하도록 solution 함수를 완성해보세요.

 

 

 

제한 사항 )

• 1 ≤ s의 길이 ≤ 200
• -1,000 < s의 원소 중 숫자 < 1,000
• s는 숫자, "Z", 공백으로 이루어져 있습니다.
• s에 있는 숫자와 "Z"는 서로 공백으로 구분됩니다.
• 연속된 공백은 주어지지 않습니다.
• 0을 제외하고는 0으로 시작하는 숫자는 없습니다.
• s는 "Z"로 시작하지 않습니다.
• s의 시작과 끝에는 공백이 없습니다.
• "Z"가 연속해서 나오는 경우는 없습니다.

 

 

 

입출력 예 )

 

 

입출력 예 설명 )

입출력 예 #1

• 본문과 동일합니다.

입출력 예 #2

• 10 + 20 + 30 + 40 = 100을 return 합니다.

입출력 예 #3

• "10 Z 20 Z 1"에서 10 다음 Z, 20 다음 Z로 10, 20이 지워지고 1만 더하여 1을 return 합니다.

입출력 예 #4, #5

설명 생략

 

 

풀이)

 

#include <string>
#include <vector>
#include <sstream>
 
using namespace std;
 
int solution(string s) {
    int answer = 0;
    
    stringstream sstream;
    sstream.str(s);
 
    // 처음 입력은 무조건 answer로 넣어주고,
    // 해당 값을 이전 값에도 입력해놓는다
    sstream >> answer;
    int num = answer;
    
    // 들어온 값을 temp에 저장
    string temp;
    while (sstream >> temp)
    {
        // 만약 "Z"라면 이전값을 빼준다
        if (temp == "Z")
        {
            answer -= num;
        }
        // 이전값 갱신 + 현재값 더해주기
        else 
        {
            int temp_num = stoi(temp);
            answer += temp_num;
            num = temp_num;
        }
    }
    
    return answer;
}

---

출처 : https://school.programmers.co.kr/learn/courses/30/lessons/120853

프로그래머스