[파이썬] 프로그래머스_문자열 계산하기

문제

 

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

프로그래머스

---

 

해설

 

split()을 사용하면 공백을 기준으로 문자를 나눠준다. 

answer에 미리 시작 숫자를 넣어놓고 i값이 연산자일 때만 맞는 계산을 하도록 해주면 답은 쉽게 구할 수 있다.

 

이렇게 안해도 문자열로 식을 입력하면 해당식을 실행한 결과값을 반환해주는 함수가 있다.

def solution(my_string):
    return eval(my_string)

이렇게 해도 답을 구할 수 있다.

---

코드

 

-파이썬

def solution(my_string):
    arr = my_string.split()
    answer = int(arr[0])

    for i in range(len(arr)):
        if arr[i] == '+':
            answer += int(arr[i+1])
        elif arr[i] == '-':
            answer -= int(arr[i+1])
    
    
    return answer