주차 요금 계산/C++

프로그래머스

풀이 방법

 

문자열을 시간 / 차량 번호 / 내역으로 잘라서 차량번호를 Key 해당 차량 번호의 시간들을 Value로 해서 map 컨테이너에 저장한다.   만약 value의 개수가 홀수라면 그 차량이 나간 기록이 records에 담겨 있지 않으므로 23:59에 출차한 것이니 value에 "23:59"를 추가한다.  이후 차량이 하루동안 몇 분 주차되어 있는지를 계산해서 기본 시간보다 짧으면 기본 요금을 아니라면 기본 요금에 단위 시간마다 단위 요금을 추가해서 answer에 담는다.

 

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

using namespace std;

int convert_min(string s) {
    int total_min = 0;
    
    int hour = stoi(s.substr(0, 2));
    int min = stoi(s.substr(3, 2));
    
    total_min = hour * 60 + min;
    
    return total_min;
}

vector<int> solution(vector<int> fees, vector<string> records) {
    vector<int> answer;
    map<string, vector<string>> m;
    
    for(int i = 0; i < records.size(); i++) {
        vector<string> v;
        
        string tmp = "";
        for(int j = 0; j < records[i].length(); j++) {
            if(records[i][j] == ' ') {
                v.push_back(tmp);
                tmp = "";
            }
            else tmp += records[i][j];
        }
        v.push_back(tmp);
        
        m[v[1]].push_back(v[0]);
        v.clear();
    }
    
    for(auto it : m) {
        vector<string> times = it.second; 
                
        if(times.size() % 2) times.push_back("23:59");
        
        int time = 0;
        for(int i = 0; i < times.size() - 1; i += 2) {
            int num1 = convert_min(times[i]);
            int num2 = convert_min(times[i+1]);
            time += num2 - num1;
        }
        
        if(time > fees[0])
            answer.push_back(fees[1] +  ceil((time - fees[0]) / (double)fees[2]) * fees[3]);
        else answer.push_back(fees[1]);
    }

    
    return answer;
}