프로그래머스 - 주차 요금 계산(Java, 자바)

문제 링크

프로그래머스

문제 풀이 방법

1. IN 일때, HashMap 에 차량 번호와 입차 시간을 분으로 변경 한 뒤 put 한다.
2. OUT 일때, HashMap 에 있는 값을 가져와서 분 차이를 계산한다.
3. 누적 시간을 관리하는 HashMap 에 차량 시간과 누적 분을 넣는다.
4. records 탐색 완료후 input HashMap에 남아 있는 값이 있다면 23:59 기준으로 다시 시간 차이를 계산해서 누적시간 관리하는 out HashMap 에 해당 값을 넣는다.
5. 누적시간으로 이제 주차 요금을 계산한다.
6. 이때 초과 주차요금을 계산해야 되는데 올림계산을 해야한다.

---

전체 코드

import java.util.*;
class Solution {
    public int[] solution(int[] fees, String[] records) {
        int stdTime = fees[0];
        int stdFee = fees[1];
        int addTime = fees[2];
        int addFee = fees[3];

        HashMap<String, Integer> input = new HashMap<>();
        HashMap<String, Integer> out = new HashMap<>();

        for (int i = 0; i < records.length; i++) {
            String[] temp = records[i].split(" ");
            int m = time(temp[0]);
            if(temp[2].equals("IN")){
                input.put(temp[1], m);
            }
            else{
                int diff = m - input.get(temp[1]);
                input.remove(temp[1]);
                out.put(temp[1], out.getOrDefault(temp[1], 0) + diff);
            }
        }

        int end = time("23:59");
        for(String time : input.keySet()){
            int diff = end - input.get(time);
            out.put(time, out.getOrDefault(time, 0) + diff);
        }

        ArrayList<String> numbers = new ArrayList<>();
        for(String num : out.keySet()){
            numbers.add(num);
        }
        Collections.sort(numbers);

        int[] answer = new int[numbers.size()];

        for (int i = 0; i < numbers.size(); i++) {
            int cur = out.get(numbers.get(i));
            int fee = 0;
            if(cur > stdTime){
                cur = cur - stdTime;
                double t = Math.ceil((double) cur / addTime);
                fee = (int) t * addFee;
            }
            answer[i] = stdFee + fee;
        }
        return answer;
    }

    public static int time(String timestamp){
        int hours = Integer.parseInt(timestamp.split(":")[0]) * 60;
        int minute = Integer.parseInt(timestamp.split(":")[1]);
        return hours + minute;
    }
}