[프로그래머스/C++] 간단한 식 계산하기

문제

문자열 binomial이 매개변수로 주어집니다. binomial은 "a op b" 형태의 이항식이고 a와 b는 음이 아닌 정수, op는 '+', '-', '*' 중 하나입니다. 주어진 식을 계산한 정수를 return 하는 solution 함수를 작성해 주세요.

 

솔루션

#include <string>
#include <vector>

using namespace std;

int solution(string binomial) {
    int answer = 0;
    for(int i = 0; i < binomial.size(); i++){
        if(binomial[i] == '+'){
            return stoi(binomial.substr(0, i-1))+stoi(binomial.substr(i+2, binomial.size()-i+1));
        }
        else if(binomial[i] == '-'){
            return stoi(binomial.substr(0, i-1))-stoi(binomial.substr(i+2, binomial.size()-i+1));
        }
        else if(binomial[i] == '*'){
            return stoi(binomial.substr(0, i-1))*stoi(binomial.substr(i+2, binomial.size()-i+1));
        }
    }
    return answer;
}

 

https://school.programmers.co.kr/learn/courses/30/lessons/181865

프로그래머스