[프로그래머스] 7주차_입실 퇴실

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

코딩테스트 연습 - 7주차

 

서로 반드시 만나는 경우는 entrance(= 입실한 사람) 중 퇴실 명부의 1순위가 있으면 바로 퇴실 했을 때, 만나는 경우를 말한다.

 

예를 들어, entrance 에 {1, 2, 3, 4, 5} 가 있고 퇴실 명부의 1순위 즉, leave 의 첫 번째 원소가 1이라고 하자.

그럼 1번이 나가면서 answer[1-1] 은 2, 3, 4, 5번이랑 반드시 만났으므로 += 4 가 된다. 그리고 1이 entrance 에서 제거해주는데 이때, 2번이 퇴실했을 경우 1번과 만났다는 정보가 없어지는 것이므로 1이 나갈 때, 2, 3, 4, 5번이 1번과 만났다는 것을 answer[i-1] (i는 2, 3, 4, 5) 에 += 1 을 해준다. 

 

def solution(enter, leave):
    answer = [0] * len(enter)
    enter.reverse()
    leave.reverse()
    entrance = set()
    
    while leave:
        if leave[-1] in entrance:
            l = leave.pop()
            entrance.remove(l)
            answer[l-1] += len(entrance)
            
            for i in entrance:
                answer[i-1] += 1
        else:
            if enter:
                entrance.add(enter.pop())
        
    return answer

print(solution([1, 3, 2], [1, 2, 3])) # [0, 1, 1]