컨트롤 제트
2024. 8. 13. 18:00ㆍ코딩테스트 입문
https://school.programmers.co.kr/learn/courses/30/lessons/120853
문제 설명
숫자와 "Z"가 공백으로 구분되어 담긴 문자열이 주어집니다. 문자열에 있는 숫자를 차례대로 더하려고 합니다. 이 때 "Z"가 나오면 바로 전에 더했던 숫자를 뺀다는 뜻입니다. 숫자와 "Z"로 이루어진 문자열 s가 주어질 때, 머쓱이가 구한 값을 return 하도록 solution 함수를 완성해보세요.
제한사항
- 1 ≤ s의 길이 ≤ 200
- -1,000 < s의 원소 중 숫자 < 1,000
- s는 숫자, "Z", 공백으로 이루어져 있습니다.
- s에 있는 숫자와 "Z"는 서로 공백으로 구분됩니다.
- 연속된 공백은 주어지지 않습니다.
- 0을 제외하고는 0으로 시작하는 숫자는 없습니다.
- s는 "Z"로 시작하지 않습니다.
- s의 시작과 끝에는 공백이 없습니다.
- "Z"가 연속해서 나오는 경우는 없습니다.
입출력 예
s | result |
"1 2 Z 3" | 4 |
"10 20 30 40" | 100 |
"10 Z 20 Z 1" | 1 |
"10 Z 20 Z" | 0 |
"-1 -2 -3 Z" | -3 |
입출력 예 설명
입출력 예 #1
- 본문과 동일합니다.
입출력 예 #2
- 10 + 20 + 30 + 40 = 100을 return 합니다.
입출력 예 #3
- "10 Z 20 Z 1"에서 10 다음 Z, 20 다음 Z로 10, 20이 지워지고 1만 더하여 1을 return 합니다.
입출력 예 #4, #5
- 설명 생략
정수와 Z로 이루어진 문자열에서 올바른 정수들만 추출해서 더하는 문제입니다.
이 문제가 다른 문제랑 다른 점이라면 음수도 포함된다는 점입니다. 보통은 입문 단계에서는 양수만 주거든요.
음수가 있기 때문에 단순히 문자열을 리스트처럼 사용하는 게 아니라, 공백을 기준으로 나눠줘야 합니다.
특정 단어를 기준으로 나누는 split() 함수를 사용해서 문자열을 나누고, Z 연산을 하면서 문제를 풀어보겠습니다.
파이썬을 사용합니다.
def solution(s):
# 숫자를 담은 배열
answer = []
# 공백을 기준으로 s를 나누어 sn으로 하나씩 가져온다.
for sn in s.split():
# sn이 Z라면 마지막 값을 제거한다.
if sn == 'Z':
answer.pop()
# 정수라면 추가한다.
else:
answer.append(int(sn))
# 배열의 합을 반환한다.
return sum(answer)
딱히 어려울 건 없고, 이 문제에 음수가 섞여있다는 것만 고려해서 split()을 사용할 수만 있으면 쉽게 풀 수 있습니다.
'코딩테스트 입문' 카테고리의 다른 글
직사각형 넓이 구하기 (0) | 2024.08.15 |
---|---|
배열 원소의 길이 (0) | 2024.08.14 |
소인수분해 (0) | 2024.08.12 |
숨어있는 숫자의 덧셈 (1) (0) | 2024.08.11 |
문자열 정렬하기 (1) (0) | 2024.08.10 |