문제 설명

주차장의 요금표와 차량이 들어오고(입차) 나간(출차) 기록이 주어졌을 때, 차량별로 주차 요금을 계산하려고 합니다. 아래는 하나의 예시를 나타냅니다.

• 요금표

• 입/출차 기록

• 자동차별 주차 요금

• 어떤 차량이 입차된 후에 출차된 내역이 없다면, 23:59에 출차된 것으로 간주합니다.
• • 0000번 차량은 18:59에 입차된 이후, 출차된 내역이 없습니다. 따라서, 23:59에 출차된 것으로 간주합니다.
• 00:00부터 23:59까지의 입/출차 내역을 바탕으로 차량별 누적 주차 시간을 계산하여 요금을 일괄로 정산합니다.
• 누적 주차 시간이 기본 시간이하라면, 기본 요금을 청구합니다.
• 누적 주차 시간이 기본 시간을 초과하면, 기본 요금에 더해서, 초과한 시간에 대해서 단위 시간 마다 단위 요금을 청구합니다.
• • 초과한 시간이 단위 시간으로 나누어 떨어지지 않으면, 올림합니다.
• • ⌈a⌉ : a보다 작지 않은 최소의 정수를 의미합니다. 즉, 올림을 의미합니다.

주차 요금을 나타내는 정수 배열 fees, 자동차의 입/출차 내역을 나타내는 문자열 배열 records가 매개변수로 주어집니다. 차량 번호가 작은 자동차부터 청구할 주차 요금을 차례대로 정수 배열에 담아서 return 하도록 solution 함수를 완성해주세요.

제한 사항

• fees의 길이 = 4

• • fees[0] = 기본 시간(분)
• • 1 ≤ fees[0] ≤ 1,439

• -fees[1] = 기본 요금(원)

• • 0 ≤ fees[1] ≤ 100,000
• • fees[2] = 단위 시간(분)
• • 1 ≤ fees[2] ≤ 1,439
• • fees[3] = 단위 요금(원)
• • 1 ≤ fees[3] ≤ 10,000

• 1 ≤ records의 길이 ≤ 1,000

• records의 각 원소는 "시각 차량번호 내역" 형식의 문자열입니다.

• • 시각, 차량번호, 내역은 하나의 공백으로 구분되어 있습니다.
• • 시각은 차량이 입차되거나 출차된 시각을 나타내며, HH:MM 형식의 길이 5인 문자열입니다.
• • HH:MM은 00:00부터 23:59까지 주어집니다.
• • 잘못된 시각("25:22", "09:65" 등)은 입력으로 주어지지 않습니다.
• • 차량번호는 자동차를 구분하기 위한, `0'~'9'로 구성된 길이 4인 문자열입니다.
• • 내역은 길이 2 또는 3인 문자열로, IN 또는 OUT입니다. IN은 입차를, OUT은 출차를 의미합니다.
• • records의 원소들은 시각을 기준으로 오름차순으로 정렬되어 주어집니다.
• • records는 하루 동안의 입/출차된 기록만 담고 있으며, 입차된 차량이 다음날 출차되는 경우는 입력으로 주어지지 않습니다.
• • 같은 시각에, 같은 차량번호의 내역이 2번 이상 나타내지 않습니다.
• • 마지막 시각(23:59)에 입차되는 경우는 입력으로 주어지지 않습니다.
• • 아래의 예를 포함하여, 잘못된 입력은 주어지지 않습니다.
• • • 주차장에 없는 차량이 출차되는 경우
• • • 주차장에 이미 있는 차량(차량번호가 같은 차량)이 다시 입차되는 경우

입출력 예

입출력 예 설명

내 정답 코드

import java.time.Duration;
import java.time.LocalTime;
import java.util.*;

class Solution {
    public List<Integer> solution(int[] fees, String[] records) {

        Map<String, LocalTime> hm = new HashMap<>(); // {주차한 차량 번호 : 입차 시간} 담을 해시맵
        Map<String, Long> sum = new HashMap<>(); // {주차장 이용 차량 번호 : 요금} 담을 해시맵


        for (int i = 0; i < records.length; i++) {
            String[] s = records[i].split(" ");
                LocalTime time = LocalTime.of(Integer.parseInt(s[0].substring(0, 2)), Integer.parseInt(s[0].substring(3)));
            if (!hm.containsKey(s[1])) { // IN(입차)이라면
                hm.put(s[1], time); // 차랑 번호 : 입차 시간
            } else { // OUT(출차)이라면
                Duration duration = Duration.between(hm.get(s[1]), time); // 입차 시간과 출차 시간간의 차이
                sum.put(s[1], sum.getOrDefault(s[1], 0L) + duration.toMinutes()); // 누적 요금 해시맵에 시간차 누적합 반영
                hm.remove(s[1]); // OUT(출차)했으니까 해시맵에서 삭제
            }
        }

        // 입차만 하고 출차하지 않은 차량들 요금 정산
        for (String carNum : hm.keySet()) {
            LocalTime time = LocalTime.of(23, 59);
            Duration duration = Duration.between(hm.get(carNum), time); // 입차 시간과 23:59간의 시간 차이 계산
            sum.put(carNum, sum.getOrDefault(carNum, 0L) + duration.toMinutes()); //누적 요금 해시맵에 시간차 누적합 반영
        }

        List<String> sort = new ArrayList<>(sum.keySet()); // 차량 번호 정렬을 위해 차량 번호를 sort 리스트에 담음
        Collections.sort(sort); // 차량 번호 담은 리스트 정렬

        List<Integer> result = new ArrayList<>(); // 결과 담을 리스트
        
        /*
        fees[0] : 기본 이용 시간
        fees[1] : 기본 요금
        fees[2] : 단위 시간
        fees[3] : 단위 요금
         */

        for (String carNum : sort) { // 차량 번호 오름차순 순서대로 for-each
            if (sum.get(carNum) <= fees[0]) { // 해당 차량의 주차장 총 이용 시간이 기본 시간 이하라면
                result.add(fees[1]); // 기본 요금 부과
            } else { // 해당 차량의 주차장 총 이용 시간이 기본 시간 초과시
                int fee = fees[1] + (int) Math.ceil((double)(sum.get(carNum) - fees[0]) / fees[2]) * fees[3]; // 기본 요금 + (초과 시간 / 단위 시간) 올림 * 단위 요금
                // (double)로 형변환 해주는 이유는 나누기 연산을 하면 소수점 이하가 버려지기 때문에 정확한 계산을 위해 형변환
                result.add(fee); // 결과 리스트에 추가

            }
        }
        return result;
    }
}

위의 코드에서는 차량 번호들을 리스트에 담고 해당 리스트를 정렬하는 식으로 주차 요금을 차량 번호에 따른 오름차순으로 반환하였지만 애초에 이진 탐색 트리를 기반으로 정렬을 유지하는 TreeMap을 사용하도록 코드를 약간만 수정한다면 정렬 과정을 생략할 수 있다.

import java.time.Duration;
import java.time.LocalTime;
import java.util.*;

class Solution {
    public int[] solution(int[] fees, String[] records) {

        Map<String, LocalTime> hm = new HashMap<>(); // {주차한 차량 번호 : 입차 시간} 담을 해시맵
        Map<String, Long> sum = new TreeMap<>(); // {주차장 이용 차량 번호 : 요금} 담고 차량 번호 오름차순 유지하는 트리맵


        for (int i = 0; i < records.length; i++) {
            String[] s = records[i].split(" ");
            LocalTime time = LocalTime.of(Integer.parseInt(s[0].substring(0, 2)), Integer.parseInt(s[0].substring(3)));
            if (!hm.containsKey(s[1])) { // IN이라면
                hm.put(s[1], time); // 차랑 번호 : 입차 시간
            } else { // OUT이라면
                Duration duration = Duration.between(hm.get(s[1]), time); // 입차 시간과 출차 시간간의 차이
                sum.put(s[1], sum.getOrDefault(s[1], 0L) + duration.toMinutes()); // 누적 요금 해시맵에 시간차 누적합 반영
                hm.remove(s[1]); // OUT(출차)했으니까 해시맵에서 삭제
            }
        }

        // 입차만 하고 출차하지 않은 차량들 요금 정산
        for (String carNum : hm.keySet()) {
            LocalTime time = LocalTime.of(23, 59);
            Duration duration = Duration.between(hm.get(carNum), time); // 입차 시간과 23:59간의 시간 차이 계산
            sum.put(carNum, sum.getOrDefault(carNum, 0L) + duration.toMinutes()); //누적 요금 해시맵에 시간차 누적합 반영
        }

        int[] result = new int[sum.size()]; // 결과 배열

        /*
        fees[0] : 기본 이용 시간
        fees[1] : 기본 요금
        fees[2] : 단위 시간
        fees[3] : 단위 요금
         */


        int idx = 0; // idx를 0으로 초기화
        for (String carNum : sum.keySet()) { // 차량 번호 오름차순 순서대로 for-each
            if (sum.get(carNum) <= fees[0]) { // 해당 차량의 주차장 총 이용 시간이 기본 시간 이하라면
                result[idx] = fees[1]; // 기본 요금 부과
            } else { // 해당 차량의 주차장 총 이용 시간이 기본 시간 초과시
                int fee = fees[1] + (int) Math.ceil((double)(sum.get(carNum) - fees[0]) / fees[2]) * fees[3]; // 기본 요금 + (초과 시간 / 단위 시간) 올림 * 단위 요금
                result[idx] = fee; // 결과 리스트에 추가
            }
            idx++; // idx 증가
        }
        return result;
    }
}