[프로그래머스] 문자열 계산하기 (C++)

문자열 계산하기 / Lv.0

---

문제  설명 )

my_string은 "3 + 5"처럼 문자열로 된 수식입니다. 문자열 my_string이 매개변수로 주어질 때, 수식을 계산한 값을 return 하는 solution 함수를 완성해주세요.

 

 

 

제한 사항 )

• 연산자는 +, -만 존재합니다.
• 문자열의 시작과 끝에는 공백이 없습니다.
• 0으로 시작하는 숫자는 주어지지 않습니다.
• 잘못된 수식은 주어지지 않습니다.
• 5 ≤ my_string의 길이 ≤ 100
• my_string을 계산한 결과값은 1 이상 100,000 이하입니다.
  • my_string의 중간 계산 값은 -100,000 이상 100,000 이하입니다.
  • 계산에 사용하는 숫자는 1 이상 20,000 이하인 자연수입니다.
  • my_string에는 연산자가 적어도 하나 포함되어 있습니다.
• return type 은 정수형입니다.
• my_string의 숫자와 연산자는 공백 하나로 구분되어 있습니다.

 

 

 

입출력 예 )

 

 

입출력 예 설명 )

입출력 예 #1

• 3 + 4 = 7을 return 합니다.

 

 

풀이)

 

#include <string>
#include <vector>
#include <sstream>
 
using namespace std;
 
int solution(string my_string) {
    int answer = 0;
    
    // 공백을 기준으로 입력받기 위해 sstream의 
    // stringstream 사용
    stringstream s;
    s.str(my_string);
    
    // 연산자 저장
    string opt;
    
    // 첫 입력은 무조건 숫자이므로 먼저 입력 받는다.
    s >> answer;
    string temp;
    
    // temp에 저장하며 계속 확인
    while(s >> temp)
    {
        // 입력받은게 연산자라면, opt에 저장하고 넘어간다.
        if (temp == "+")
        {
            opt = "+";
            continue;
        }
        
        if(temp == "-")
        {
            opt = "-";
            continue;
        }
        
        // 입력받은게 숫자라면 opt에 따라 계속해서
        // answer에 답을 누적시킨다.
        if(opt == "+")
        {
            answer += stoi(temp);
        }
        else if (opt == "-")
        {
            answer -= stoi(temp);
        }
    }
    
    return answer;
}

---

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

프로그래머스