[C++] 프로그래머스 2018 KAKAO BLIND RECRUITMENT [1차] 셔틀버스

(1시간 40)

HH:MM형태의 시간을 분으로 치환(int로 변환) 후 우선순위 큐에 집어넣었다.

제일 늦은 시간을 구하는 거니까 마지막 운행 차량 전까지는 탑승 가능한 인원에 맞춰 제거해주고

마지막 운행에서는 탑승 인원이 빌 경우 마지막 운행시간이 도착 시간 중 제일 늦은 시각이고

탑승 인원이 가득 찼을 경우 제일 마지막에 탄 사람에서 -1 한게 도착 시간 중 제일 늦은 시간이다.

 

#include <string>
#include <vector>
#include <queue>

using namespace std;

string solution(int n, int t, int m, vector<string> timetable) {
	priority_queue<int, vector<int>, greater<int>> pq;
	vector<int> bus;
	int answer = 0;

	// 크루 대기열 시간 정렬하기
	for (int i = 0; i < timetable.size(); i++) {
		int hour = atoi(timetable[i].substr(0, 2).c_str());
		int minute = atoi(timetable[i].substr(3, 2).c_str());
		int time = hour * 60 + minute;
		pq.push(time);
	}

	// 버스 운영시간 저장
	int startTime = 9 * 60;
	for (int i = 0; i < n; i++) {
		bus.push_back(startTime + t * i);
	}

	// 마지막 운영 시간 전까지 대기열 지우기
	int idx;
	for (idx = 0; idx < bus.size() - 1; idx++) {
		for (int j = 0; j < m; j++) {
			if (!pq.empty()) {
				if (bus[idx] >= pq.top()) {
					pq.pop();
				}
			}
		}
	}
    
    // 마지막 버스가 가득 찼는지
    bool isEmpty = true;
    
	// 마지막 운영 시간
	for (int j = 0; j < m; j++) {
		if (!pq.empty()) {
			if (bus[idx] >= pq.top()) {
                answer = pq.top();
				pq.pop();
            }else {
                answer = bus[idx];
                isEmpty = false;
                break;
		    }
		}else {
			answer = bus[idx];
            isEmpty = false;
			break;
		}
	}
    
    // 버스가 가득 찼을 경우 제일 마지막에 탄 사람보다 1 작은 경우가 정답
    if(isEmpty) answer = answer - 1;

	// 시간 변환
	string ansHour = answer / 60 > 9 ? to_string(answer / 60) : '0' + to_string(answer / 60);
	string ansMinute = answer % 60 > 9 ? to_string(answer % 60) : '0' + to_string(answer % 60);
	string ansStr = ansHour + ':' + ansMinute;

	return ansStr;
}