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

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

프로그래머스

풀이

요금표와 입출차 기록이 주어졌을 때, 주차 요금을 계산해야 하는 문제입니다.

최종 출력은 번호판 오름차순으로 총 요금을 answer에 추가하여 리턴해야 합니다.

 

문자열 파싱 및 map자료구조 해싱을 사용해 문제를 풀이했습니다.

차량이 입차 했을 때의 시간을 기록하기 위해 map 자료구조 변수 m_for_record 를 선언합니다. 또한, OUT이 주어졌을 때, 주차 시간을 누적하기 위한 m_for_calc_time 변수를 선언합니다. 

 

입력변수 records 를 순회하며 IN 이 주어졌다면 m_for_record에 기록해둔 후, OUT이 주어지면 현재 시간 time에서 입차 시간을 빼주어 m_for_calc_time에 기록해둡니다. 

이때, map의 value는 값이 없는 경우도 0이기 때문에, 입차 시간이 00:00인 경우와 구별하기 위해 이때만 -1로 기록해둡니다. 또한 출차했다면 비어있음을 의미하는 0으로 다시 초기화해둡니다. 

 

다음으로, m_for_record 변수를 순회하며 value가 존재하거나 -1인 경우 아직 출차하지 않은 경우이기 때문에 23:59을 출차시간으로 적용하여 다시 m_for_calc_time에 기록합니다.

 

요금 계산의 경우 주어진 예시대로 코드를 구현하면 됩니다. 

map 자료구조는 key값 기준 기본 오름차순 정렬이기 때문에 별도 정렬 없이 총 요금을 answer에 추가한 후 리턴하면 됩니다. 

#include <string>
#include <vector>
#include <iostream>
#include <map>
#include <cmath>

using namespace std;

vector<int> solution(vector<int> fees, vector<string> records) {
    vector<int> answer;
    map<string, int> m_for_record;
    map<string, int> m_for_calc_time;
    int max_time = 23*60 + 59;
    double base_time = fees[0];
    double base_fee = fees[1];
    double unit_time = fees[2];
    double unit_fee = fees[3];
    double total_fee;
    
    for (auto i : records) {
        int time = stoi(i.substr(0, 2))*60 + stoi(i.substr(3, 2));
        string v_num = i.substr(6, 4);
        string in_or_out = i.substr(11);
        
        if (in_or_out == "IN") {
            // 입차 시간 기록
            m_for_record[v_num] = time; 
            if (time == 0) m_for_record[v_num] = -1; // 00:00에 들어온 경우 특수 케이스
        }
        else {
            // OUT 인 경우 총 시간 계산 및 m_for_record 초기화
            m_for_calc_time[v_num] += (time - m_for_record[v_num]);
            m_for_record[v_num] = 0; // 출차
        }
    }
    // 출차 하지 않은 차량 총 시간 계산
    for (auto i : m_for_record) {
        if (i.second || i.second == -1) {
            // 출차 하지 않은 경우 또는 특수 케이스는 23:59로 출차
            if (i.second == -1)
                m_for_calc_time[i.first] += max_time;
            else 
                m_for_calc_time[i.first] += (max_time - i.second);
        }
    }
    // 요금 계산
    for (auto i : m_for_calc_time) {
        // 기본 시간 이하
        if (i.second <= base_time) {
            answer.push_back(base_fee); // 기본 요금
        }
        else {
            total_fee = base_fee + ceil((i.second - base_time) / unit_time) * unit_fee;
            answer.push_back(total_fee);
        }
    }
    
    return answer;
}