[프로그래머스] 호텔 방 배정(Python)

https://programmers.co.kr/learn/courses/30/lessons/64063

코딩테스트 연습 - 호텔 방 배정

딕셔너리를 활용해서 해당 방 선택 시 몇 번으로 이동하면 되는지 저장하면 된다.

 

def solution(k, room_number):
    answer = []
    room = {}    
    for num in room_number:
        index = num
        visit = [index]
        while index in room:
            index = room[index]
            visit.append(index)
        answer.append(index)
        for i in visit:
            room[i] = index + 1
    return answer