[프로그래머스] 2. 문자열 계산하기 - Python

프로그래머스 문제 풀이 : 문자열 계산하기

 

• 프로그래밍 언어 : Python
• 난이도 : Level 0

 

문제집 링크 : http://aladin.kr/p/szPra

내일은 코딩테스트 with 파이썬

 

프로그래머스 URL : 

[도서실습] 내일은 코딩테스트 with 파이썬(자료구조와 알고리즘의 기초부터 실전까지)

---

■ 작성 코드

def solution(my_string):
    
    my_string = my_string.split(" ")
    ans = int(my_string[0])
    
    for i in range(len(my_string)):
        if my_string[i] =="+":
            ans += int(my_string[i+1])
        elif my_string[i] =="-":
            ans -= int(my_string[i+1])
    
    return ans

 

1. my_string.split(" ")을 통해서 문자열을 리스트로 변환
2. ans는 my_string의 첫 번째 요소 → "3 + 4" 라면 3이 된다.
3. 반복문을 통해서 +, - 일 때, 각각 다음 요소를 더하거나 빼준다.