[프로그래머스/Programmers] 코딩 테스트 입문 > 문자열 계산하기 (파이썬 Python)

문제

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

프로그래머스

 

내 제출1 (정답)

def solution(my_string):
    # eval 함수를 사용하여 문자열로 된 수식을 평가
    return eval(my_string)

- eval() 함수 사용 : 주어진 제한사항에 따르면 잘못된 수식은 주어지지 않으니 eval()을 사용해도 문제가 없음.

파이썬에서 수식을 문자열로 평가하기 위한 여러 방법이 있지만, 가장 간단한 방법은 eval() 함수를 사용하는 것. eval() 함수는 문자열로 된 수식을 평가하여 그 결과를 반환한.

다만, eval() 함수는 사용에 주의가 필요하며, 외부 입력을 그대로 평가할 경우 보안 문제가 발생할 수 있습니다.

 

내 제출2 (정답)

def solution(my_string):
    my_list = my_string.split(" ")
    total = int(my_list[0])
    operator = None
    
    for element in my_list[1:]:
        if element in ["+", "-"]:
            operator = element
        else:
            if operator == "+":
                total += int(element)
            elif operator == "-":
                total -= int(element)
                
    return total

 

- 수식을 공백을 기준으로 분리하여 리스트로 만든다.

- 첫 번째 숫자를 초기 값으로 설정.

- 리스트를 순회하면서 연산자와 숫자를 처리.

- 연산자를 만나면 operator 변수에 저장하고, 숫자를 만나면 total에 누적 계산.

 

오답이었던 제출

def solution(my_string):
    my_list = my_string.split(" ")
    value1 = int(my_list[0])
    value2 = int(my_list[2])
    if my_list[1] == '+' :
        return value1 + value2
    else:
        return value1 - value2

 

- 오답 이유 : 두 개의 숫자와 하나의 연산자가 있는 경우에 대해서는 올바르게 작동하지만, 연산자가 여러 개 있는 더 복잡한 수식에는 대응하지 못한다.

- 반례 : "3 + 5 - 2"

- 나는 "2 + 5"같이 그냥 숫자 2개, 연산자 한 개인 줄 알았음..;