알고리즘

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

Dev_Lee 2023. 3. 14. 00:42

문제 설명

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


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

입출력 예sresult
"1 2 Z 3" 4
"10 20 30 40" 100
"10 Z 20 Z 1" 1
"10 Z 20 Z" 0
"-1 -2 -3 Z" -3

입출력 예 설명

입출력 예 #1

  • 본문과 동일합니다.

입출력 예 #2

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

입출력 예 #3

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

우선 공백으로 문자열을 나눠서 벡터에 집어넣는 것 부터 해야한다.

stringstream을 이용해서 words 벡터에 문자열을 넣어줌

words 벡터에서 for 문을 돌리며 Z일 경우에 answer에서 앞의 문자를 빼주고

Z가 아닐경우에는 그냥 더해주면 됨

string 벡터에 있는 문자를 int로 변환해서 덧셈 뺄셈을 하고 싶으면

아스키코드표를 보고 변환해서 연산해줘도 되는데

이 문제의 경우엔 숫자 범위가 -1000 to 1000이라 연산이 복잡해질거같아서

stoi를 사용해서 string을 int로 변환해주고 연산했음

#include <string>
#include <vector>
#include <sstream>

using namespace std;

int solution(string s) {
    int answer = 0;
    
    stringstream ss(s);
    vector<string> words;
    string word;
    while (getline(ss, word, ' ')){
        words.push_back(word);
    }
    
    for(int i = 0; i < words.size(); i++)
    {
        if(words[i] != "Z")
        {
            answer = answer + stoi(words[i]);
        }
        else 
        {
            answer = answer - stoi(words[i-1]);
        }
    }
    
   
    return answer;
}