[Programmers] 주차 요금 계산 (Java)

1.  문제 링크

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

2.  문제

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

• 요금표
  
  
• 입/출차 기록
  
  
• 자동차별 주차 요금
  
  
• 어떤 차량이 입차된 후에 출차된 내역이 없다면 23:59에 출차된 것으로 간주합니다.
  • 0000번 차량은 18:59에 입차된 이후, 출차된 내역이 없습니다.
• 00:00부터 23:59까지의 입/출차 내역을 바탕으로 차량별 누적 주차 시간을 계산하여 요금을 일괄로 정산합니다.
• 누적 주차 시간이 기본 시간 이하라면, 기본 요금을 청구합니다.
• 누적 주차 시간이 기본 시간을 초과하면, 기본 요금에 더해서, 초과한 시간에 대해서 단위 시간마다 단위 요금을 청구합니다.
  • 초과한 시간이 단위 시간으로 나누어 떨어지지 않으면, 올림합니다.
  • ⌈a⌉ : a보다 작지 않은 최소의 정수를 의미합니다. 즉, 올림을 의미합니다.
  주차 요금을 나타내는 정수 배열 fees, 자동차의 입/출차 내역을 나타내는 문자열 배열 records가 매개변수로 주어집니다. 차량 번호가 작은 자동차부터 청구할 주차 요금을 차례대로 정수 배열에 담아서 return 하도록 solution 함수를 완성해주세요.

3.  제한사항

• 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)에 입차되는 경우는 입력으로 주어지지 않습니다.
  • 아래의 예를 포함하여, 잘못된 입력은 주어지지 않습니다.
  • 주차장에 없는 차량이 출차되는 경우
  • 주차장에 이미 있는 차량(차량번호가 같은 차량)이 다시 입차되는 경우

입출력 예

4.  소스코드

import java.util.*;

class Solution {
    public int[] solution(int[] fees, String[] records) {
        TreeMap<String, Integer> map = new TreeMap<>();
        HashMap<String, String> inTimes = new HashMap<>();
        HashMap<String, Integer> times = new HashMap<>();
        for(String record : records) {
            String[] record_content = record.split(" ");
            String time = record_content[0];
            String carNumber = record_content[1];
            if(record_content[2].equals("IN")) {
                inTimes.put(carNumber, time);
            } else if(record_content[2].equals("OUT")) {
                String[] time_parse = time.split(":");
                String[] inTime = inTimes.get(carNumber).split(":");
                inTimes.put(carNumber, "-1");
                int hour = Integer.parseInt(time_parse[0]) - Integer.parseInt(inTime[0]);
                int min = Integer.parseInt(time_parse[1]) - Integer.parseInt(inTime[1]);
                int totalTime = hour * 60 + min;
                if(times.containsKey(carNumber)) {
                    times.put(carNumber, totalTime + times.get(carNumber));
                } else {
                    times.put(carNumber, totalTime);
                }
            }
        }
        for(String key : inTimes.keySet()) {
            if(!inTimes.get(key).equals("-1")) {
                String[] inTime = inTimes.get(key).split(":");
                int hour = 23 - Integer.parseInt(inTime[0]);
                int min = 59 - Integer.parseInt(inTime[1]);
                int totalTime = hour * 60 + min;
                if(times.containsKey(key)) {
                    times.put(key, times.get(key) + totalTime);
                } else {
                    times.put(key, totalTime);
                }
            }
        }
        for(String key : times.keySet()) {
            int totalTime = times.get(key);
            int fee = fees[1];
            totalTime -= fees[0];
            if(totalTime > 0) {
                int extra_time = (int)Math.ceil((double)totalTime / fees[2]);
                fee += extra_time * fees[3];
            }
            map.put(key, fee);
        }
        int[] answer = new int[map.size()];
        int idx = 0;
        for(String key : map.keySet()) {
            answer[idx] = map.get(key);
            idx++;
        }
        return answer;
    }
}

5.  접근

1. 입/출차한 각 차들의 누적 주차 시간을 계산합니다.
   • 각 차의 입차 시간을 의미하는 HashMap inTimes를 생성하고 각 차의 누적 주차 시간을 의미하는 HashMap times를 생성합니다.
   • 주어진 입/출차 기록을 이용하여 입차하였을 때는 inTimes에서 해당 차량에 대한 시간을 업데이트해주고 출차했을 시에는 -1로 설정합니다.
   • 출차할 때는 inTimes에 있는 입차 시간을 이용하여 시간을 계산하고 이를 times에 누적해갑니다.
   • 만약 모든 입/출차 기록을 이용하였는데도 아직 inTimes에 -1이 아닌 차가 있다면 23:59이 지났을 때도 차가 출차하지 않은 상황이기 때문에 해당 차들에 대해서는 23:59을 출차 시간으로 하여 시간을 계산해 times에 누적해줍니다.
2. 각 차들의 누적 주차 시간을 이용하여 주차 요금을 계산합니다.
   • 각 차들의 누적 주차 시간에서 기본 시간을 빼주고 기본 요금을 부과합니다.
   • 만약 누적 주차 시간에서 기본 시간을 뺐을 때 0 또는 음수가 된다면 더이상 부과할 요금이 없기 때문에 해당 차량의 주차 요금은 기본 요금이 됩니다.
   • 만약 누적 주차 시간에서 기본 시간을 뺐을 때 양수라면 단위 시간으로 남은 시간을 나눠주고 소숫점 아래로 숫자가 존재한다면 일의 자리까지 올림을 진행하여 몇 번 단위 요금을 부과해야하는지 구합니다.
   • (구한 횟수 * 단위 요금)을 기본 요금에 더한 값이 최종 주차 요금이 됩니다.
3. 차량 번호가 작은 자동차부터 청구할 주차 요금을 차례대로 정수 배열에 담아 return합니다.
   • 2번에서 각 차량의 주차 요금을 계산한 후에 TreeSet에 넣어주었는데 TreeSet 자료구조에 따라 key를 기준으로 오름차순으로 정렬되기 때문에 이를 answer 정수 배열에 담아 return합니다.