[프로그래머스] 주차 요금 계산

문제

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

• 요금표

기본 시간(분)	기본 요금(원)	단위 시간(분)	단위 요금(원)
180	5000	10	600

• 입/출차 기록

시각	차량 번호	내역
05:34	5961	입차
06:00	0000	입차
06:34	0000	출차
07:59	5961	출차
07:59	0148	입차
18:59	0000	입차
19:09	0148	출차
22:59	5961	입차
23:00	5961	출차

• 자동차별 주차 요금

차량 번호	누적 주차 시간(분)	주차 요금(원)
0000	34 + 300 = 334	5000 + ⌈(334 - 180) / 10⌉ x 600 = 14600
0148	670	5000 + ⌈(670 - 180) / 10⌉ x 600 = 34400
5961	145 + 1 = 146	5000

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

입출력 예

fees	records	result
[180, 5000, 10, 600]	["05:34 5961 IN", "06:00 0000 IN", "06:34 0000 OUT", "07:59 5961 OUT", "07:59 0148 IN", "18:59 0000 IN", "19:09 0148 OUT", "22:59 5961 IN", "23:00 5961 OUT"]	[14600, 34400, 5000]
[120, 0, 60, 591]	["16:00 3961 IN", "16:00 0202 IN", "18:00 3961 OUT", "18:00 0202 OUT", "23:58 3961 IN"]	[0, 591]
[1, 461, 1, 10]	["00:00 1234 IN"]	[14841]

문제 출처

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

생각

• 차량 번호가 적은 자동차부터 청구 요금을 구해야 하므로, 차량 번호를 key 값으로 하는 TreeMap을 사용
• IN일 때는 차량의 입차 기록을 담는다
• OUT일 때는 차의 주차 시간(출차 - 입차)을 구한뒤 Tree Map에 업데이트. 처리된 입차 기록은 제거
• 루프가 끝난 뒤, 처리 안된 입차 기록 즉 출차 안된 차량들은 23:59에 출차되었다고 가정 후 TreeMap 업데이트
• 각 차량 별 누적 주차시간을 사용하여 주차 요금을 계산하여 return
  • 이 때, 추가 요금 계산 시 올림은 Math.ceil로 처리

코드

import java.util.*;

class Solution {
    public int[] solution(int[] fees, String[] records) {
        
        Map<String,String> timeStampPerCar = new HashMap<>();
        Map<String,Integer> timePerCar = new TreeMap<>();
        
        for (String record: records){
            String[] parts = record.split(" ");
            String time = parts[0];
            String carNum = parts[1];
            String inOut = parts[2];
            if (inOut.equals("IN")){
                timeStampPerCar.put(carNum, time);
            } else if (inOut.equals("OUT")){
                timePerCar.put(carNum, timePerCar.getOrDefault(carNum, 0) + calculateTimeDiff(time, timeStampPerCar.get(carNum)));
                timeStampPerCar.remove(carNum);
            }
        }
        
        for (Map.Entry<String,String> entry: timeStampPerCar.entrySet()){
            timePerCar.put(entry.getKey(), timePerCar.getOrDefault(entry.getKey(), 0) + calculateTimeDiff("23:59", entry.getValue()));
        }
        
        int[] answer = new int[timePerCar.size()];
        int idx = 0;
        for (Map.Entry<String,Integer> entry: timePerCar.entrySet()){
            int parkingTime = entry.getValue();
            answer[idx++] = (parkingTime < fees[0]) ? fees[1] :  fees[1] +  (int) Math.ceil((parkingTime - fees[0]) / (double) fees[2]) * fees[3];
        }
        
        return answer;
    }
    
    public int calculateTimeDiff(String outTime, String inTime){
        
        String[] outParts = outTime.split(":");
        String[] inParts = inTime.split(":");
        
        int outHour = Integer.parseInt(outParts[0]);
        int outMinute = Integer.parseInt(outParts[1]);
        int inHour = Integer.parseInt(inParts[0]);
        int inMinute = Integer.parseInt(inParts[1]);

        return (outHour - inHour) * 60 + outMinute - inMinute;

    }
}

다른 사람의 풀이

import java.util.*;

class Solution {

    public int timeToInt(String time) {
        String temp[] = time.split(":");
        return Integer.parseInt(temp[0])*60 + Integer.parseInt(temp[1]);
    }
    public int[] solution(int[] fees, String[] records) {

        TreeMap<String, Integer> map = new TreeMap<>();

        for(String record : records) {
            String temp[] = record.split(" ");
            int time = temp[2].equals("IN") ? -1 : 1;
            time *= timeToInt(temp[0]);
            map.put(temp[1], map.getOrDefault(temp[1], 0) + time);
        }
        int idx = 0, ans[] = new int[map.size()];
        for(int time : map.values()) {
            if(time < 1) time += 1439;
            time -= fees[0];
            int cost = fees[1];
            if(time > 0)
                cost += (time%fees[2] == 0 ? time/fees[2] : time/fees[2]+1)*fees[3];

            ans[idx++] = cost;
        }
        return ans;
    }
}

• 입차 할 때 -1을 곱하고, 출차할 때 1을 곱해 누적 시간을 한 번에 계산한 방식이 너무 좋았다 👍👍
• 다만 출차하지 않은 차량을 23:59분에 출차한다고 가정 할 때 1439를 time에 더해주는 것보다 만들어놓은 timeToInt() 메서드를 사용했으면 더 좋았을 것 같다는 생각이 들었다