[프로그래머스]문자열 계산하기, eval() 함수

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

"3 + 4"

result

7

 

내 풀이(오답)

def solution(my_string):
    answer = 0

    if my_string[2] == '+':
        answer = int(my_string[0]) + int(my_string[-1])
    elif my_string[2] == '-':
        answer = int(my_string[0]) - int(my_string[-1])
    
    return answer

 

 

• 여러 자리 수 처리 불가:
  • my_string[0]과 my_string[-1]을 사용해 연산을 수행하고 있기 때문에, 숫자가 한 자리 수가 아닌 경우 올바르게 처리되지 않는다.
  • 예를 들어, "12 + 34"라는 입력이 들어오면 my_string[0]은 '1'이고 my_string[-1]은 '4'로 잘못된 연산이 수행된다.
• 공백이 다양한 위치에 있을 때:
  • 현재 코드에서는 공백의 위치나 개수에 민감하게 작동하기 때문에, "3 + 5" 외에 " 3 + 5", "3+5" 등의 입력이 들어오면 작동하지 않는다.

 

 

 

 

.eval() 함수 :  문자열로 표현된 파이썬 표현식을 계산해 주는 함

def solution(my_string):
    return eval(my_string)

 

• eval() 함수는 사용이 간단하지만, 사용자가 입력한 임의의 코드를 실행할 수 있으므로 보안에 취약할 수 있다. 사용자가 입력을 직접 제어할 수 있는 상황에서는 eval()을 사용하는 것을 피해야 한다.
• 이 문제의 경우, 수식이 단순한 덧셈과 뺄셈으로 한정되어 있으므로 eval()을 사용해도 무방