[Programmers] 코딩테스트 입문 - 문자열 계산하기

문제

my_string은 "3 + 5"처럼 문자열로 된 수식입니다. 문자열 my_string이 매개변수로 주어질 때, 수식을 계산한 값을 return 하는 solution 함수를 완성해주세요.

 

 

제한사항

1. 연산자는 +, -만 존재합니다.
2. 문자열의 시작과 끝에는 공백이 없습니다.
3. 0으로 시작하는 숫자는 주어지지 않습니다.
4. 잘못된 수식은 주어지지 않습니다.
5. 5 ≤ my_string의 길이 ≤ 100
6. my_string을 계산한 결과값은 1 이상 100,000 이하입니다.
   • 계산에 사용하는 숫자는 1 이상 20,000 이하인 자연수입니다.
   • my_string에는 연산자가 적어도 하나 포함되어 있습니다.
   • my_string의 중간 계산 값은 -100,000 이상 100,000 이하입니다.
7. return type 은 정수형입니다.
8. my_string의 숫자와 연산자는 공백 하나로 구분되어 있습니다.

 

 

 

---

 

 

작성 답안

def solution(my_string):
    
    string_list = my_string.split(' ')


    num_list = [int(x) for x in string_list if x.isdigit()]
    opr_list = [x for x in string_list if x.isdigit() == False]

    answer = num_list[0]

    for i in range(1, len(num_list)):
        if opr_list[i-1] == '+':
            answer+= num_list[i]
        else:
            answer-= num_list[i]
    return answer

 

 

 

 

작성 답안에 대한 풀이 

수식을 문자열 타입으로 입력값을 받는 형태의 문제이다. 여기서의 핵심은 "1000 + 200 - 3" 처럼 연산 기호가 여러 개인 경우를 어떻게 처리하느냐에 달린 것 같았다.

 

어떻게 풀이 할 지 고민하다 보니, 위의 수식처럼 숫자가 3개이면 연산 기호는 무조건 2개인 것을 깨닫게 되었다.

우선 문자열을 공백 기준으로 split 하여 리스트에 담았다. 그리고 정답을 리턴할 최종적으로 answer 변수에 리스트에 있는 변수 중 첫번쨰 값을 무조건 담았다.

 

그리고 각각의 남은 값을 계산하면서 그에 대응하는 연산 기호를 가져와 확인한 후 연산하였다.

 

다만 이 방법은 우선.. 반복문을 사용한다는 점이 좋지 않은 방식 같았다. 아마.. 리스트 컴프리헨션으로도 처리를 할 수 있을 것이다. 

 

 

 

 

 

다른 사람 풀이

신기해서 가져와보았다.. 이런 아이디어는 참고하면 좋을 것 같다. 

다른 풀이들도 살펴보았는데, 파이썬 내부적으로 eval() 함수를 사용하면 바로 입력된 문자열 수식을 계산할 수 있다. 

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

https://school.programmers.co.kr/learn/courses/30/lessons/120902/solution_groups?language=python3

프로그래머스