조앤의 기술블로그

[프로그래머스 / 카카오] (틀림)호텔 방 배정 본문

Programming/프로그래머스

[프로그래머스 / 카카오] (틀림)호텔 방 배정

쬬앤 2020. 4. 4. 11:05

[1차코드] - 실패

테스트케이스는 통과했으나 효율성에서 통과하지 못했다.

import java.util.*;
class Solution {
    public long[] solution(long k, long[] room_number) {
        long[] answer = new long[room_number.length];
        boolean[] room = new boolean[(int)k+1];
        
        Arrays.fill(room, false);
        
        int i = 0;
        for(long r : room_number){
            if(room[(int)r] == false){
                // 방이 비어있다. 
                room[(int)r] = true;
                answer[i++] = (int)r;
            }else{
                for(int j = (int)r ; j < room.length; j++){
                    if(room[j] == false){
                        room[j] = true;
                        answer[i++] = j;
                        break;
                    }
                }
            }
        }
        return answer;
    }
}