[코딩테스트 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의 숫자와 연산자는 공백 하나로 구분되어 있습니다.

입출력 예

my_string	result
"3 + 4"		7

입출력 예 설명

입출력 예 #1
3 + 4 = 7을 return 합니다.

---

내 풀이

Java

class Solution {
    public int solution(String my_string) {
        int answer = 0;
        String[] str = my_string.split(" ");
        
        
        String op = "";
        for(int i = 0; i < str.length; i++){
            if(str[i].equals("+")){
                op = "+";
            }else if(str[i].equals("-")){
                op = "-";
            }else{
                if(!op.equals("")){
                    int n = Integer.parseInt(str[i]);
                    answer = op.equals("+") ? answer + n: answer - n;
                }else{
                    answer = Integer.parseInt(str[i]);
                }
            }
        }
        
        return answer;
    }
}

C

#include <stdio.h>
#include <stdbool.h>
#include <stdlib.h>

// 파라미터로 주어지는 문자열은 const로 주어집니다. 변경하려면 문자열을 복사해서 사용하세요.
int solution(const char* my_string) {
    int answer = 0;    
    int op = -1;
    int n = 0;
    for(int i = 0; i < strlen(my_string); i++){
        while(my_string[i] >= '0' && my_string[i] <= '9'){
            n *= 10;
            n += my_string[i] - '0';
            i++;
        }
        
        if(op) answer = answer + n;
        else answer = answer - n;
        n = 0;
        
        if(my_string[i] == '+') op = 1;
        if(my_string[i] == '-') op = 0;
        
    }
    
    
    return answer;
}

Python

def solution(my_string):
    answer = 0
    
    list = my_string.split(" ")
    op = ""
    for s in list:
        if s == '+' or s == '-':
            op = s
        else:
            if op != '':
                n = int(s)
                if op == '+':
                    answer = answer + n
                else:
                    answer = answer - n
            else:
                answer = int(s)
               
    
    return answer