NIRVANA
[level 0] 문자열 계산하기 본문
1. 문제
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의 숫자와 연산자는 공백 하나로 구분되어 있습니다.
2. 문제 풀이 방법
1) answer와 oper를 각각 0과 ""로 선언한다.
2) my_string으로 for 반복문을 시작한다.
3) 만약 i가 연산자라면 oper를 해당 연산자로 변경한다.
4) 만약 i가 숫자라면 연산자에 따라 값을 더하거나 빼준다.
-이때, 만약 연산자가 공백이라면 첫번째 숫자이므로 answer의 값을 i로 변경한다. (사실 +해도 상관 없을 것 같긴함..)
5) answer return
def solution(my_string):
answer = 0
oper = ""
for i in my_string.split():
if i in ["+", "-"]:
oper = i
else:
if oper == "+":
answer += int(i)
elif oper == "-":
answer -= int(i)
else:
answer = int(i)
return answer
다른 문제 풀이 방법
1) 문자열에서 "-"를 "+ -"로 변경한다.
2) 문자열을 "+"연산자 기준으로 나누어 리스트로 만든 뒤, 해당 리스트로 반복문을 진행한다.
-이렇게 되면 4 + 3 → 4 + (-3)이 됨
3) 해당 리스트의 숫자들을 int형으로 변경한 뒤, answer에 더한다.
def solution(my_string):
answer = 0
for num in my_string.replace("-", "+ -").split("+"):
answer += int(num)
return answer
'Coding test(Python3) > Programmers' 카테고리의 다른 글
[level 2] 전화번호 목록 (0) | 2024.06.24 |
---|---|
[level 0] 자릿수 더하기 (0) | 2024.01.16 |
[level 2] 귤 고르기 (0) | 2023.11.13 |
[level 2] 멀리 뛰기 (1) | 2023.11.12 |
[level 2] N개의 최소공배수(최소공배수, 최대공약수 복습) (0) | 2023.11.08 |