Code/프로그래머스

[2019 카카오 개발자 겨울 인턴십] 호텔 방 배정

코-빗 2024. 3. 24. 16:23
728x90
import java.util.HashMap;
import java.util.Map;

class Solution {
    public long[] solution(long k, long[] room_number) {
        long[] answer = new long[room_number.length];

        Map<Long, Long> map = new HashMap<>();
        
        for(int i = 0; i < room_number.length; i++) {
        	answer[i] = getRoomNumber(map, room_number[i]);
        }
        
        return answer;
    }
    
    static long getRoomNumber(Map<Long, Long> map, long n) {
    	if(map.get(n) == null) {
    		map.put(n , n + 1);
    		return n;
    	}
    	else {
    		long res = getRoomNumber(map, map.get(n));
    		map.replace(n , res + 1);
    		return res;
    	}
    }
}
728x90