[code] - 카카오 호텔 방 배정 python

#호텔 방 배정

def solution(k, room_number):
    a = [0 for _ in range(k+1)]
    b = []
    for i in range(len(room_number)):
        if a[room_number[i]] == 0 :
            a[room_number[i]] = room_number[i]
            b.append(room_number[i])
        elif a[room_number[i]] != 0 :
            for j in range(1,len(room_number)-room_number[i]+1):
                if a[room_number[i]+j] == 0 :
                    a[room_number[i]+j] = room_number[i]+j
                    b.append(room_number[i]+j)
                    
                    
                    
    answer = b
    return answer

위의 코드로 실행 시킬 경우, 테스트 case는 정답이지만 실제 채점에서는 박날남. 나중에 다시 뜯어보자.

def solution(k, room_number):
    a = []
    for i in range(len(room_number)):
        if room_number[i] not in a :
            a.append(room_number[i])
        
        elif room_number[i] in a :
            for j in range(len(room_number)):
                if room_number[i]+j not in a :
                    a.append(room_number[i]+j)
                    break  
           
    return a

두번째 시도한 코드. 효율성 박살남