문자열 계산하기

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

프로그래머스

[문제 설명]

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

[나의 풀이]

def solution(my_string):
    return eval(my_string)

⭐eval() 함수는 문자열 형태로 제공된 수식을 파이썬 코드처럼 계산해 주는 함수이다.

 

[다른 풀이]

def solution(my_string):
    return sum(int(i) for i in my_string.replace(' - ', ' + -').split(' + '))

' - '는 ' + -'로 변경하여 ' + '로 split한 다음 sum하는 방법 

 

def solution(my_string):
    s = my_string.split()
    answer = int(s[0])
    for i in range(1,len(s),2):
        if s[i]=='+':
            answer+=int(s[i+1])
        else:
            answer-=int(s[i+1])
    return answer

for i in range(1,len(s),2) 코드는 i를 인덱스로 설정하기 위함

if s[i] == '+'로 코드 진행

def solution(my_string):
    my_string = my_string.split()
    answer = 0
    operator = ''
    for m in my_string:
        if m == '+' or m == '-':
            operator = m
        else:
            if operator == '+':
                answer += int(m)
            elif operator == '-':
                answer -= int(m)
            elif operator == '':
                answer = int(m)
    return answer

my_string을 split한 값을 통해 answer와 operator값을 업데이트하는 방식으로 답 도출

 

def solution(my_string):
    num=[]
    pnm=['+']
    for i in my_string.split():
        if i=='+' or i=='-':
            pnm.append(i)
        else:
            num.append(i)
    answer = 0
    for pair in zip(pnm,num):
        if pair[0]=='+':
            answer+=int(pair[1])
        else:
            answer-=int(pair[1])
    return answer

⭐zip() 함수를 이용해서 부호와 숫자의 pair를 만들어줌