[Programmers] 호텔 방 배정

#include <string> #include <vector> #include <map> using namespace std; typedef long long ll; vector<long long> solution(long long k, vector<long long> num) { vector<long long> answer; map<ll, int> roomMap; map<ll, int> ::iterator it; for (int i = 0; i < num.size(); i++) { if (roomMap[num[i]] == 0) { roomMap[num[i]] = i+1; //0을 피하기 위해 answer.push_back(num[i]); } else { //원하는 방이 이미 배정된 경우 ll idx = num[i]+1; while(1) { if (roomMap[idx] == 0) break; idx++; } roomMap[idx] = i+1; answer.push_back(idx); } } return answer; }