728x90
[ 코딩테스트 연습 - 프로그래머스 Lv.2 ]
- 문제명 : 주차 요금 계산 [ 2022 KAKAO BLIND RECRUITMENT ]
- 사용 언어 : Python3
- 주요 사용 핵심 : dictionary를 사용하여 문제 해결.
문제 설명
주차장의 요금표와 차량이 들어오고(입차) 나간(출차) 기록이 주어졌을 때, 차량별로 주차 요금을 계산하려고 합니다.
- 어떤 차량이 입차된 후에 출차된 내역이 없다면, 23:59에 출차된 것으로 간주합니다. 0000번 차량은 18:59에 입차된 이후, 출차된 내역이 없습니다. 따라서, 23:59에 출차된 것으로 간주합니다.
- 00:00부터 23:59까지의 입/출차 내역을 바탕으로 차량별 누적 주차 시간을 계산하여 요금을 일괄로 정산합니다.
- 누적 주차 시간이 기본 시간이하라면, 기본요금을 청구합니다.
- 누적 주차 시간이 기본 시간을 초과하면, 기본 요금에 더해서, 초과한 시간에 대해서 단위 시간마다 단위 요금을 청구합니다.
- 초과한 시간이 단위 시간으로 나누어 떨어지지 않으면, 올림합니다.
- ⌈a⌉ : a보다 작지 않은 최소의 정수를 의미합니다. 즉, 올림을 의미합니다.
주차 요금을 나타내는 정수 배열 fees, 자동차의 입/출차 내역을 나타내는 문자열 배열 records가 매개변수로 주어집니다. 차량 번호가 작은 자동차부터 청구할 주차 요금을 차례대로 정수 배열에 담아서 return 하도록 solution 함수를 완성해 주세요.
▼ 실행 코드 ▼
import math
def solution(fees, records):
answer = []
in_dict = dict() # 입차 시간을 저장하는 딕셔너리
cumul_time = dict() # 누적 주차 시간을 저장하는 딕셔너리
for entry in records:
parts = entry.split() # 공백을 기준으로 문자열 분리
time , number, status = parts[0], parts[1], parts[2]
h, m = map(int, time.split(":"))
time = h * 60 + m # 시간을 분 단위로 변환
if status == "IN":
# 입차 시간을 기록
in_dict[number] = time
else:
# 출차 시간에 따라 누적 주차 시간을 계산하고, 입차 기록 삭제
cumul_time[number] = cumul_time.get(number,0) + time - in_dict[number]
del in_dict[number]
# 주차장에 남아있는 차량들의 누적 주차 시간 계산
for number in in_dict:
cumul_time[number] = cumul_time.get(number, 0) + (23 * 60 + 59) - in_dict[number]
# 각 차량의 주차 요금 계산
for number, time in sorted(cumul_time.items()):
if time > fees[0]:
# 누적 주차 시간이 기본 시간을 초과하는 경우
# 초과한 시간에 대한 추가 요금 계산
answer.append(fees[1] + math.ceil((time - fees[0]) / fees[2]) * fees[3])
else:
# 기본 시간 이내의 주차 시간이면 기본 요금만 추가
answer.append(fees[1])
return answer
▽ 결과 확인 ▽
학습 회고
: 처음에는 cumul_time 딕셔너리에서 number 키에 대한 값에 접근할 때 cumul_time[number]와 같이 작성하여 해당 키가 없을 때 발생하는 KeyError를 고려하지 않았다. 그 결과, 딕셔너리에 존재하지 않는 차량 번호에 대한 누적 주차 시간을 조회할 때 발생하는 예외로 코드가 중단되었다.
문제를 해결하기 위해 cumul_time.get(number, 0)을 사용하여 해당 키가 없을 경우 기본값으로 0을 반환하도록 수정하여, 예외를 방지하고 코드가 정상적으로 실행될 수 있도록 보완하였다.
728x90
반응형
'Algorithm' 카테고리의 다른 글
[Algorithm] BAEKJOON 12789번. 도키도키 간식드리미 (Python) (0) | 2024.01.18 |
---|---|
[Algorithm] BAEKJOON 2504번. 괄호의 값 (Python) (0) | 2024.01.17 |
[Algorithm] 타겟 넘버 - 프로그래머스 Lv.2 (0) | 2024.01.12 |
[Algorithm] 깊이/너비 우선 탐색 (DFS/BFS) (0) | 2024.01.12 |
[자료구조] 이진 탐색 트리(BST, Binary Search Tree) (0) | 2023.12.29 |
댓글