‘파이썬 알고리즘 인터뷰(박상길, 책만)’ 참조
1. 문제 - (카카오 18년 기출) - 다트 게임
- 카카오톡 게임별의 하반기 신규 서비스로 다트 게임을 출시하기로 했다. 다트 게임은 다트판에 다트를 세 차례 던져 그 점수의 합계로 실력을 겨루는 게임으로, 모두가 간단히 즐길 수 있다. 갓 입사한 무지는 코딩 실력을 인정받아 게임의 핵심 부분인 점수 계산 로직을 맡게 되었다. 다트 게임의 점수 계산 로직은 아래와 같다.
- 다트 게임은 총 3번의 기회로 구성된다.
- 각 기회마다 얻을 수 있는 점수는 0점에서 10점까지이다.
- 점수와 함께 Single(S), Double(D), Triple(T) 영역이 존재하고 각 영역 당첨 시 점수에서 1제곱, 2제곱, 3제곱 (점수1 , 점수2 , 점수3 )으로 계산된다.
- 옵션으로 스타상() , 아차상(#)이 존재하며 스타상(*) 당첨 시 해당 점수와 바로 전에 얻은 점수를 각 2배로 만든다. 아차상(#) 당첨 시 해당 점수는 마이너스된다.
- 스타상()은 첫 번째 기회에서도 나올 수 있다. 이 경우 첫 번째 스타상(*)의 점수만 2배가 된다. (예제 4번 참고)
- 스타상()의 효과는 다른 스타상()의 효과와 중첩될 수 있다. 이 경우 중첩된 스타상() 점수는 4배가 된다. (예제 4번 참고)
- 스타상(*)의 효과는 아차상(#)의 효과와 중첩될 수 있다. 이 경우 중첩된 아차상(#)의 점수는 -2배가 된다. (예제 5번 참고)
- Single(S), Double(D), Triple(T)은 점수마다 하나씩 존재한다.
- 스타상(*), 아차상(#)은 점수마다 둘 중 하나만 존재할 수 있으며, 존재하지 않을 수도 있다.
- 0~10의 정수와 문자 S, D, T, *, #로 구성된 문자열이 입력될 시 총점수를 반환하는 함수를 작성하라.
2. 풀이 - 문자열 처리
1) 정수 처리
문자열을 처리하는 문제다.
다트를 던지는 횟수(3번)만큼 덧셈이 이뤄지므로, 총 세개의 점수가 더해지는 셈이다.
리스트를 생성하여 세개의 점수를 삽입하고 sum()함수로 그 답을 구하는 구조로 전체를 구성한다.
문자열(dartResult)을 대상으로 for 반복문을 실행하여 개별 문자열 하나하나를 추출한다.
추출한 문자열(s)가 정수일 경우, 미리 만든 리스트(nums)에 해당 문자열의 정수값(int(s))을 입력하는 방식이다.
정수값 앞에 ‘ nums[-1] * 10 ’이 더해진 이유는 10점이 입력값으로 들어오는 경우 때문이다.
다트 점수가 10점일 경우, for문의 문자열 처리 과정에서 각 자리 숫자가 개별적으로(‘1’ → ‘0’) 처리된다.
따라서 일의 자리 수 ‘0’을 처리할 때, 십의 자리수 ‘1’에 대한 자릿수 올림을 계산해야 한다.
nums = [0]
for s in dartResult:
if s.isdigit(): # s가 정수일 때
nums[-1] = nums[-1] * 10 + int(s)
2) 연산식 처리
거듭제곱 연산(Single, Double, Triple)과 곱셉(스타상 및 아차상)을 처리한다.
거듭제곱 연산은 1제곱, 2제곱, 3제곱을 처리한다.
스타상은 방금 던진 다트의 점수와 이전에 던진 다트의 점수에 ‘2’를 곱하고,
아차상은 방금 던진 다트 점수에 ‘-1’을 곱한다.
리스트에 들어있는 정수 값에 각각의 연산을 적용시키면 된다.
주의할 점으로 스타상의 경우 2차례 시행을 대상으로 ‘x2’처리를 해야하며,(#1)
스타상이 맨 첫 시행에서 발생할 경우에 대한 예외 조건을 설정해야 한다. (#2)
전체 코드는 아래와 같다.
def solution(dartResult):
answer = 0
nums = [0]
for s in dartResult:
if s == 'S':
nums[-1] **= 1
nums.append(0)
elif s == 'D':
nums[-1] **= 2
nums.append(0)
elif s == 'T':
nums[-1] **= 3
nums.append(0)
# 1. 이전 값과 그 이전 값 모두 2배 처리
elif s == '*':
nums[-2] *= 2
#2. 아차상이 맨 처음 시행에서 발생할 경우
if len(nums) > 2:
nums[-3] *= 2
elif s == '#':
nums[-2] *= (-1)
# 자릿수 올림: 입력값으로 10이 들어온 경우, 문자열 '1'과 '0'을 나누어 처리
else:
nums[-1] = nums[-1] * 10 + int(s)
return sum(nums)
'자료구조 & 알고리즘' 카테고리의 다른 글
카카오 코딩 테스트 18년 기출 - 3. 캐시 (0) | 2022.07.18 |
---|---|
카카오 코딩 테스트 18년 기출 - 1. 비밀 지도 (0) | 2022.07.13 |
[파이썬] 그래프 - 코스 스케줄[리트코드 207] (0) | 2022.07.13 |
[파이썬] 그래프 - 일정 재구성(리트코드 332) (0) | 2022.07.12 |
[파이썬] 조합의 합 - 리트코드(39) (0) | 2022.07.12 |