[프로그래머스] 주차 요금 계산 (JAVA 풀이)

 

문제 접근

 

https://school.programmers.co.kr/learn/courses/30/lessons/92341

프로그래머스

 

 

문제를 살펴보면 규칙에 따라 주차 요금을 계산하는 구현 문제임을 쉽게 파악할 수 있음

=> 규칙들을 살펴보고 미리 주석으로 구현해야할 내용들을 정리하고 풀이를 시작함

 

문제 풀이

 

우선 입력이 잘못들어오는 경우가 없다고 조건이 주어져있으므로 예외처리는 크게 신경쓰지 않아도됨

또한, 기본 요금이 분 단위이기에 더 쉬운 시간 계산을 위해 입차, 출차 시간을 분으로 변환하고 모든 로직을 수행함

 

우선 누적(하나의 차량이 여러번 IN, OUT을 할 수 있음) 주차 시간을 기준으로 비용을 계산하기 때문에 누적 추가 시간을 관리하는 Map과 입차 (현재 주차되어 있는 차)를 관리하는 Map을 선언하였음

 

만약 IN일 경우는 단순히 parkingMap에 차량 번호와 입차 시간을 저장해줌

 

OUT일 경우는 해당 차량 번호의 입차시간을 parkingMap에서 가져온 후 출차 시간에서 빼줘서 누적 주차 시간(단위 : 분)을 누적 주차 시간을 관리하는 timeMap에 저장함

(이 때 OUT은 출차를 의미하므로 현재 주차된 차를 저장하는 parkingMap에서 remove해줘야함)

 

이렇게 for문을 돌면서 주어진 입/출차 기록을 다 반영했다면 parkingMap에 남은 끝까지 출차되지 않은 차량들은 23:59를 기준으로 출차시켜야함

 

for문으로 parkingMap의 차량 번호에 접근하며 위에서 말한 OUT 로직에서 출차 시간만 23:59로 하여 동일한 로직을 돌림

 

마지막으로 차량 번호순으로 오름차순 정렬한 상태에서 주차 요금 결과를 반환해야하므로 TreeMap(key기준 오름차순 정렬되는 Map)을 선언한 후 요금 계산 로직을 돌리면됨

=> 기본 요금을 초기값으로 설정하고 해당 차량의 총 누적 주차 시간이 기본 시간을 넘어가게되면 해당 시간에 따른 요금을 추가해줌

 

전체 코드

 

import java.util.*;

class Solution {
    public int[] solution(int[] fees, String[] records) {
        int basicTime = fees[0];
        int basicFee = fees[1];
        int unitTime = fees[2];
        int unitFee = fees[3];

        // 1. 입차 차량 관리 Map (key: 차량번호, value: 입차시간(분))
        Map<String, Integer> parkingMap = new HashMap<>();
        // 2. 누적 주차 시간 관리 Map (key: 차량번호, value: 누적시간(분))
        Map<String, Integer> timeMap = new HashMap<>();

        for (String record : records) {
            String[] arr = record.split(" ");
            String[] timeArr = arr[0].split(":");
            int time = Integer.parseInt(timeArr[0]) * 60 + Integer.parseInt(timeArr[1]);
            String carNumber = arr[1];
            String type = arr[2];

            if (type.equals("IN")) {
                parkingMap.put(carNumber, time);
            } else { // OUT
                // 입차 시간 가져오기
                int inTime = parkingMap.get(carNumber);
                // 주차 시간 = 출차 시간 - 입차 시간
                int parkingDuration = time - inTime;
                
                // 해당 차량의 누적 주차 시간에 합산
                timeMap.put(carNumber, timeMap.getOrDefault(carNumber, 0) + parkingDuration);
                
                // 출차했으므로 입차 목록에서 제거
                parkingMap.remove(carNumber);
            }
        }

        // 3. 출차 기록이 없는 차량 처리 (23:59 출차로 간주)
        int lastTime = 23 * 60 + 59;
        for (String carNumber : parkingMap.keySet()) {
            int inTime = parkingMap.get(carNumber);
            int parkingDuration = lastTime - inTime;
            timeMap.put(carNumber, timeMap.getOrDefault(carNumber, 0) + parkingDuration);
        }

        // 4. 누적 시간을 기준으로 최종 요금 계산
        // 차량 번호를 기준으로 정렬하기 위해 TreeMap 사용
        Map<String, Integer> feeMap = new TreeMap<>();
        for (String carNumber : timeMap.keySet()) {
            int totalTime = timeMap.get(carNumber);
            
            // 기본 요금
            int fee = basicFee;
            
            // 기본 시간 초과 시 추가 요금 계산
            if (totalTime > basicTime) {
                // 초과한 시간을 단위 시간으로 나누고 올림 처리
                int overTime = totalTime - basicTime;
                int count = (int) Math.ceil((double) overTime / unitTime);
                fee += count * unitFee;
            }
            feeMap.put(carNumber, fee);
        }

        // 5. 정렬된 결과를 배열로 변환하여 반환
        int[] answer = new int[feeMap.size()];
        int i = 0;
        for (int fee : feeMap.values()) {
            answer[i++] = fee;
        }

        return answer;
    }
}